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，您应该传递您感兴趣的列名列表。