从 csv 中读取 Pandas 中没有标签的列

Question

我有以下格式的 csv 文件：

15.55   28.65   2
14.9    27.55   2
14.45   28.35   2
14.15   28.8    2
13.75   28.05   2

我想阅读它并将其保存为 X,Y, Clusters。

我尝试了以下但我没有得到任何数据

data = pd.read_csv(filename)

X = column(data, 0)
X = np.asarray(X)
Y = column(data,1)
Y = np.asarray(Y)

Answer 1

使用header=None并指定names参数：

from io import StringIO

x = """15.55   28.65   2
14.9    27.55   2
14.45   28.35   2
14.15   28.8    2
13.75   28.05   2"""

# replace StringIO(x) with 'file.csv'
df = pd.read_csv(StringIO(x), delim_whitespace=True,
                 header=None, names=['X', 'Y', 'Clusters'])

print(df)

       X      Y  Clusters
0  15.55  28.65         2
1  14.90  27.55         2
2  14.45  28.35         2
3  14.15  28.80         2
4  13.75  28.05         2

然后根据需要使用df['X'],df['Y']或df['Clusters']。在大多数情况下，没有额外需要转换到列表或数组，虽然这些操作是可能的：

df['X'].tolist()  # convert series to list
df['X'].values    # convert series to NumPy array