(Python 3)如何从 CSV 文件中每行具有可变数量的值的列中提取数据?

(Python 3)如何从 CSV 文件中每行具有可变数量的值的列中提取数据?

我有一个包含电影数据的 csv 文件。在“类型”列中,属于多个类型的电影有多个条目,这些条目使用“|”进行拆分。我想计算属于任何特定类型的电影数量。

文件截图

答案1

这是使用 Python 获取特定类型电影数量的工作版本。它必须根据您的具体用途进行调整。我只是导入了 txt 文件,将其拆分为列表列表,然后在类型列中搜索给定的字符串。

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")

相关内容