pandas - 如何将 numpy 数组附加到 pandas 数据帧

Question

我正在尝试将 3x2 numpy 数组附加到现有数据帧。像这样的东西：

import pandas as pd
import numpy as np
df = pd.Dataframe({"A": [0,0,0], "B": [1,1,1]})
arr = np.arange(6).reshape(3, 2)

df[["C", "D"]] = arr  # NOPE!

我怎样才能让它工作？

标签： pandasnumpy

Answer 1

concat在将数组转换为数据框时使用：

df = pd.concat([df, pd.DataFrame(arr, columns=["C", "D"])], axis=1)

---

   A  B  C  D
0  0  1  0  1
1  0  1  2  3
2  0  1  4  5