簡體   English   中英

如何結合多索引和數據透視在熊貓中創建真值表?

[英]How to combine Multi-index and pivot to create a truth table in Pandas?

我有以下Pandas DataFrame,其中Z列可以具有未知數量的可能值,例如A,B,C,D,E ...:

X Y Z
-----
1 1 A
1 1 B
1 1 C
1 2 A
2 1 B
2 1 C
2 2 B

我想對Z列進行透視,對X,Y列進行多索引,以創建一個真值表,如下所示:

X Y A B C
---------
1 1 T T T
1 2 T F F
2 1 F T T
2 2 F T F

如何在熊貓中做到這一點?

您還可以創建值列Trues然后用pivot_table()

df["Value"] = True   
df.pivot_table("Value", ["X", "Y"], "Z", fill_value=False).reset_index()

在此處輸入圖片說明

下面的代碼(幾乎)做到了:

pd.get_dummies(df,'Z').groupby(['X','Y']).max()

以下內容更接近您要求的確切輸出:

(pd.get_dummies(df,'Z')
 .groupby(['X','Y'])
 .max()
 .astype(bool)
 .applymap(lambda s: str(s)[0])
 .rename(columns=lambda v: v.replace('Z_','')))

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM