python pandas 变量赋值

Question

我对 python pandas 变量赋值有疑问:我有一个像

的元组

asd=('prostate1.csv','dtime','status1','age','hg','sz','sg','pf','rx')

读取文件就可以了:

prostate_dataset=pd.read_csv(asd[0])

但是修剪数据集并不能无缝工作:

prostate_dataset=prostate_dataset[[for x in asd[1:]]]

我想要得到的是这样的:

prostate_dataset=prostate_dataset[[asd[1],asd[2],asd[3],asd[4],asd[5],asd[6],asd[7],asd[8]]]

我已经尝试过:

act='\',\''.join(asd[1:])
prostate_dataset=prostate_dataset[[act]]

但它不起作用，因为反斜杠符号仍然包含在内

提前致谢

Answer 1

把它变成一个列表来过滤你的df:

prostate_dataset=prostate_dataset[list(asd[1:])]

应该有效:

In [157]:
asd=('prostate1.csv','dtime','status1','age','hg','sz','sg','pf','rx')
list(asd[1:])

Out[157]:
['dtime', 'status1', 'age', 'hg', 'sz', 'sg', 'pf', 'rx']

这里要理解的是，对元组进行切片将返回具有切片范围内的值的元组，但要索引 df，您应该传递您感兴趣的列名称列表。