从python中的csv文件中读取数据

Question

我正在尝试从虚拟csv文件中提取数据以在tensorflow内部使用。 虚拟数据只有2列：X（单个特征列）和Y（预期输出）。

X     Y 
11.0 13.0
23.0 33.3 
...  ...  and so on

现在我正在阅读这样的数据：

import pandas as pd

dummy_data = pd.read_csv("dummy_data.csv", sep=",")
inputX = dummy_data.loc[:, 'X'].values
np.reshape(inputX, [11, 1])

我正在重塑numpy数组，因为我需要稍后使用线性回归进行矩阵乘法，但我想问的是从csv数据中提取列的正确方法是什么？有没有更好的方法直接将csv数据提取到张量对象？

Answer 1

无需重新塑造或使用.loc或.values：

inputX = dummy_data[['X']]

（注意列表清单[[]]！）