(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");
Другие вопросы по тегам