如何将数据框中的两列添加到 np.array

Question

我正在尝试添加我的 df 数据框中包含的两列，但错误显示“DataFrame”对象不可调用，这是我的代码：

df = pd.read_csv('data.txt')

df.drop(['id'], 1, inplace=True)

#print(df)

x = np.array(df.drop(['S', 'T'], axis=1))

y = np.array(df(['S', 'T']))

Answer 1

错误是正确的。一个pd.DataFrame对象是不是调用。使用pd.DataFrame.__getitem__或其语法糖[]与系列列表：y = np.array(df[['S', 'T']])或者，更好的是，使用pd.DataFrame.values：y = df[['S', 'T']].values