Как добавить массив NumPy в кадр данных Pandas

Question

Я пытаюсь добавить массив NxPy 3x2 к существующему фрейму данных. Примерно так:

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!

Как мне заставить это работать?

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

Answer 2

Это не сработало, потому что вам нужно передать df:

arr = pd.DataFrame(np.arange(6).reshape(3, 2))

df[["C", "D"]] = arr #YEP

#Output
    A   B   C   D

0   0   1   0   1

1   0   1   2   3

2   0   1   4   5