【问题标题】:Pivot a column containing list旋转包含列表的列
【发布时间】:2021-12-16 12:33:55
【问题描述】:

我有一个名为 dfGenres 的 DataFrame,其中一列 Genres 看起来像:

DataFrame 的代码:

dfGenres = pd.DataFrame({'Genres':["[Action, Romance, Mystery, Animal]", "[Bromance, Drama, Mystery]", "[Horror, Monsters, Shounen, Action]"]})

Pivot 表示我希望拥有相同数量的行,但在每一列中,列表中的每个唯一元素,如果该行包含列表中的流派,则填充 1,否则填充 0。

我尝试了get_dummies:

pd.get_dummies(
    dfGenres["Genres"].str.split('[\[,\]]',expand=True).stack()
    )

但我得到了太多行(我只想要一个而不是 multiIndex):

【问题讨论】:

    标签: python pandas dataframe pivot dummy-variable


    【解决方案1】:

    试试str.get_dummies:

    dfGenres["Genres"].str.strip('[]').str.get_dummies(sep=', ')
    

    【讨论】:

      猜你喜欢
      • 2022-07-06
      • 2019-04-14
      • 2022-12-03
      • 2020-10-07
      • 1970-01-01
      • 1970-01-01
      • 2019-07-26
      • 2022-10-23
      • 1970-01-01
      相关资源
      最近更新 更多