Как я могу добавить два моих столбца в моем фрейме данных в 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