如何將 dataframe 轉換為 0 和 1 的 ndarray？

Question

我有一個 Pandas dataframe 看起來像這樣：

col_1 col_2
a     4
a     3
b     2
c     2
d     1
b     4
c     1

我需要將它轉換為 NumPy 二維數組數組，其中每個二維數組對應於其中一個字母。 例如，如果 'a' 沒有與 1 和 2 一起出現，而是與 3 和 4 一起出現，則對應的二維數組應該類似於[0, 0, 1, 1] 。 所以在這個例子中我需要：

[[0, 0, 1, 1], [0, 1, 0, 1], [0, 1, 0, 0], [1, 0, 0, 1]]

做這個的最好方式是什么？

Answer 1

這是crosstab的一種方式

l = pd.crosstab(df.col_1,df.col_2).values.tolist()
Out[23]: [[0, 0, 1, 1], [0, 1, 0, 1], [1, 1, 0, 0], [1, 0, 0, 0]]