Как я могу создать вектор в фрейме данных pandas из других атрибутов фрейма данных?

Question

У меня есть такой фрейм данных pandas, и я хочу сгенерировать другой атрибут, содержащий элементы строки в элементе векторного типа.Например:

+--------+--------+
|  Atr1  |  Atr2  |
+--------+--------+
|  1     |  A     |
+--------+--------+
|  2     |  B     |
+--------+--------+
|  4     |  C     |
+--------+--------+

Как я могу получить это:

+--------+--------+----------+
|  Atr1  |  Atr2  | features |
+--------+--------+----------+
|  1     |  A     | [1, A]   |
+--------+--------+----------+
|  2     |  B     | [2, B]   |
+--------+--------+----------+
|  4     |  C     | [4, C]   |
+--------+--------+----------+

Answer 1

Я думаю, вам нужно:

df['features'] = df.values.tolist()

print(df)
   Atr1 Atr2 features
0     1    A   [1, A]
1     2    B   [2, B]
2     4    C   [4, C]

Если у вас есть несколько столбцов и вы хотите выбрать определенные столбцы, то:

df = pd.DataFrame({"Atr1":[1,2,4],"Atr2":['A','B','C'],"Atr3":['x','y','z']})
print(df)
   Atr1 Atr2 Atr3
0     1    A    x
1     2    B    y
2     4    C    z

#Selecting Atr2 and Atr3 columns
df['features'] = df[['Atr2','Atr3']].values.tolist()
print(df)

   Atr1 Atr2 Atr3 features
0     1    A    x   [A, x]
1     2    B    y   [B, y]
2     4    C    z   [C, z]