有没有办法在实例化的行中引用一个对象?
请参阅以下示例: 我想在读取之后删除csv文件的第一列(按索引)(通常pd.to_csv将索引输出为第一列):
df = pd.read_csv(csvfile).drop(self.columns[[0]], axis=1)
我理解自我应该放在对象上下文中,但它在这里描述了我打算做的事情。
(当然,在两个单独的行中完成此操作非常有效。)
答案 0 :(得分:2)
假设您知道数据集中的列总数以及要删除的索引 -
a = range(3)
a.remove(1)
df = pd.read_csv('test.csv', usecols = a)
这里3是总列数,我想删除第2列。您可以直接编写要使用的列索引
答案 1 :(得分:1)
一种方法是使用pd.DataFrame.iloc
:
import pandas as pd
from io import StringIO
mystr = StringIO("""col1,col2,col3
a,b,c
d,e,f
g,h,i
""")
df = pd.read_csv(mystr).iloc[:, 1:]
# col2 col3
# 0 b c
# 1 e f
# 2 h i