如何从数据帧的其他属性在熊猫数据帧中创建向量？

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]