(Python 3) Как извлечь данные из столбца с переменным количеством значений в каждой строке в файле CSV?
У меня есть CSV-файл, содержащий данные о фильмах. В столбце "Жанры" фильм, который относится к нескольким жанрам, содержит несколько записей, которые разделяются с помощью "|". Я хочу посчитать количество фильмов, которые подпадают под тот или иной жанр.
ht tps://ask-ubuntu.ru/images/af54f66789721b77fc579ae91d962f1251a03b71.png
1 ответ
Вот рабочая версия подсчета количества фильмов с определенным жанром в Python. Он должен быть адаптирован для вашего конкретного использования. Я просто импортировал текстовый файл, разделил его на список списков, а затем искал в столбце жанра заданную строку.
txt = open('file.csv');
contents = txt.read()
contents = contents.split('\n');
new = []
for line in contents:
new.append(line.split(','));
def searchGenre(contents, genreName):
count = 0;
for line in contents:
if genreName in line[genreColumnIndex]:
count += 1;
print(line);
print(count, " movies of that genre.")
print("searching for horror");
searchGenre(contents, "horror");