如何使用 pandas 在特定列中的 csv 文件中查找特定單詞

Question

我想在 csv 文件中搜索特定單詞並計算有多少單詞，我正在使用 pandas 使用 usecols 獲取特定列並使用 str.find 搜索單詞，但它只是返回整個列

def read(searchitem): 
  lst = ["author"] 
  df=pd.read_csv('data.csv',usecols=lst)
  df = df["author"].str.find(searchitem)
  print(df)
  
read('IMoRT')

Answer 1

嘗試這個：

df["author"].str.count(searchitem).sum()

編輯：

據我了解，您對兩個非常不同的事物使用相同的變量名。 它有效，但最佳實踐不建議這樣做。

def read(searchitem): 
  lst = ["author"] 
  df=pd.read_csv('data.csv',usecols=lst)
  countWord = df["author"].str.count(searchitem).sum()
  print(countWord)

Answer 2

我實際上會導入 csv 並使用 DictReader。 代碼如下所示：

import csv

with open("csv-file.csv", "r") as csv_file:
    csv_reader = csv.DictReader(csv_file)
    
    word_count = 0
    for line in csv_reader:
        if line["author"] == searchitem:
            word_count += 1