我目前有一个任务,需要下载CSV主文件,删除任何其中列A-列B <=0且列C等于给定的短语的行。我希望创建一个程序来:
- 导入CSV文件
- 删除所有列A - 列B <= 0 的行
- 要求输入以过滤列C中的一个或多个短语
- 将CSV导出到新文件中
import pandas as pd
file = read_csv("sourcefile.csv")
file['NewColumn'] = file['A'] - file['B']
file = file[file.NewColumn > 0]
columns = ['ColumnsIWantToRemove']
file.drop(columns, inplace=True, axis=1)
phrases = input('What phrases are you filtering for? ')
file = file[file.C = phrases]
file.to_csv('export.csv')
我的问题是,如何筛选包含多个短语的C列?我想让程序接受一个或多个短语,并仅显示C列值等于其中之一的行。任何指导都将是惊人的。谢谢!