在csv读取后,熊猫掉落第一列

时间:2018-03-30 10:16:52

标签: python pandas dataframe

有没有办法在实例化的行中引用一个对象?

请参阅以下示例: 我想在读取之后删除csv文件的第一列(按索引)(通常pd.to_csv将索引输出为第一列):

df = pd.read_csv(csvfile).drop(self.columns[[0]], axis=1)

我理解自我应该放在对象上下文中,但它在这里描述了我打算做的事情。

(当然,在两个单独的行中完成此操作非常有效。)

2 个答案:

答案 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