Как я могу преобразовать панду DataFrame в представление в виде массива?

Question

My Y is

       AdoptionSpeed
0                  2
1                  0
2                  3
3                  2
4                  2
5                  2
6                  1
7                  3
8                  1
9                  4
...              ...
14987              0
14988              2
14989              4
14990              3
14991              4
14992              3

[14993 rows x 1 columns]

Я хочу преобразовать это в нечто вроде ...

        0   1   2   3
0       1   1   0   0
1       0   0   0   0
2       1   1   1   0
....
14991   1   1   1   1

Так что в основном все записи с 2 становятся [1 1 0 0] и такна.

Answer 1

Бен - твой друг !!:

df = pd.DataFrame({'AdoptionSpeed': np.random.randint(16, size=10)})
df = df.AdoptionSpeed.apply(lambda x: pd.Series(list(bin(x)[2:]))).fillna(0).astype(int)
print(df)

вывод:

   0  1  2  3
0  1  1  1  1
1  1  0  1  0
2  1  1  1  0
3  1  0  1  0
4  1  0  0  0
5  1  0  0  1
6  1  1  0  1
7  1  1  1  0
8  1  1  1  0
9  1  0  0  1

Answer 2

То же, что и мое предыдущее решение

pd.DataFrame([np.ones(x) for x in df['AdoptionSpeed']]).fillna(0).astype(int)
Out[70]: 
   0  1  2  3
0  1  1  0  0
1  0  0  0  0
2  1  1  1  0
3  1  1  0  0
4  1  1  0  0
5  1  1  0  0
6  1  0  0  0
7  1  1  1  0
8  1  0  0  0
9  1  1  1  1

---

Или больше похоже на numpy широковещание: должно быть быстрее, чем для цикла

x=df.AdoptionSpeed.max()
pd.DataFrame((df.AdoptionSpeed.values[:,None]>np.arange(x)).astype(int))
Out[81]: 
   0  1  2  3
0  1  1  0  0
1  0  0  0  0
2  1  1  1  0
3  1  1  0  0
4  1  1  0  0
5  1  1  0  0
6  1  0  0  0
7  1  1  1  0
8  1  0  0  0
9  1  1  1  1