python - Pandas 数据帧 : extract unique component as columns

Question

我从 Bloomberg 中提取了 30 年来 SP500 的所有每日股票成分，并将其放入数据框中，其中索引为日期，行成分为股票。

现在，我需要使用日期作为索引、列作为股票来转换此数据框，以便再次将其传递到 Bloomberg API 以检索所有股票详细信息

这是我需要做的示例:

我目前有这个:

pd.DataFrame([['A','B',np.nan],['A','B','C'],['B','C','D']])

   0  1    2
0  A  B  NaN
1  A  B    C
2  B  C    D

我希望得到这个:

pd.DataFrame([[True,True,False,False],[True,True,True,False],[False,True,True,True]],columns=['A','B','C','D'])
       A     B      C      D
0   True  True  False  False
1   True  True   True  False
2  False  True   True   True

非常感谢您的帮助

Answer 1

您可以使用pd.get_dummies:

pd.get_dummies(df.stack()).max(level=0).astype(bool)

输出:

       A     B      C      D
0   True  True  False  False
1   True  True   True  False
2  False  True   True   True