我正在读取目录中的.txt文件，并希望删除包含某些特定字符串的列。
for file in glob.iglob(files + '.txt', recursive=True):
cols = list(pd.read_csv(file, nrows =1))
df=pd.read_csv(file,header=0, skiprows=0, skipfooter=0, usecols =[i for i in cols if i.str.contains['TRIVIAL|EASY']==False])
当我这样做时
df = pd.read_csv（file，header = 0，skiprows = 0，skipfooter = 0，usecols = [i for i> in cols如果i.str.contains ['PASS'] == True]）
AttributeError：'str'对象没有属性'str'
我需要解决tp问题的哪一部分？
无需单独阅读标题，您可以将调用传递给usecols。检查是否'EASY'还是'TRIVIAL'不在列名。
exclu = ['EASY', 'TRIVIAL'] # Any substring in this list excludes a column
usecols = lambda x: not any(substr in x for substr in exclu)
df = pd.read_csv('test.csv', usecols=usecols)
print(df)
HARD MEDIUM
0 2 4
1 6 8
2 1 1
test.csvTRIVIAL,HARD,EASYfoo,MEDIUM
1,2,3,4
5,6,7,8
1,1,1,1
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句