我有一个csv文件作为下面的图片
我正在尝试查找以字母A和G开头的任何单词或我想要的任何列表
但是我的代码会返回任何错误提示我在做什么错？
这是我的代码
if len(sys.argv) == 1:
print("please provide a CSV file to analys")
else:
fileinput = sys.argv[1]
wdata = pd.read_csv(fileinput)
print( list(filter(startswith("a","g"), wdata)) )
最佳答案
将Series.str.startswith与转换列表一起使用到元组，并通过DataFrame.loc与boolean indexing进行过滤：
wdata = pd.DataFrame({'words':['what','and','how','good','yes']})
L = ['a','g']
s = wdata.loc[wdata['words'].str.startswith(tuple(L)), 'words']
print (s)
1 and
3 good
Name: words, dtype: object