如何将数据框中的两个列添加到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