如何在有/没有 Pandas 的情况下在 Python 中重新排序数组？

Question

为了搜索产品和类别之间的相关性以及下一个可视化（热图），我需要使用 Python 重新排序数组，使用/不使用 Pandas 或其他库：

Book Name, Category 1, Category 2, Category 3, Django 101 Python Web-Dev Beginner ROR Guide Rails Web-Dev Intermediate Laravel PHP Web-Dev Intermediate

进入那个：

Book Name, Python, Web-Dev, Beginner, Rails, PHP, Intermediate Django 101 True True True False False, False ROR Guide False True False False False, True Laravel False True False False True, True

有没有办法做到这一点？数据存入.csv文件，由pandas.read_csv()读取

Answer 1

这可以使用 Pandas 中的get_dummies 函数来完成。

df = pd.DataFrame({'Book Name': ['Django 101', 'ROR Guide', 'Laravel'], 'Category 1': ['Python', 'Rails', 'PHP'], 'Category 2': ['Web-Dev']*3, 'Category 3': ['Beginner', 'Intermediate', 'Intermediate']})

dummies = pd.concat([pd.get_dummies(df[c]) for c in df.columns[1:]], axis=1)
df_new = pd.concat([df['Book Name'], dummies], axis=1)

>>> df_new
    Book Name  PHP  Python  Rails  Web-Dev  Beginner  Intermediate
0  Django 101    0       1      0        1         1             0
1   ROR Guide    0       0      1        1         0             1
2     Laravel    1       0      0        1         0             1

或者您可以将 DataFrame 的索引重置为书的名称：

df.set_index('Book Name', inplace=True)
df_new = pd.concat([pd.get_dummies(df[c]) for c in df], axis=1)
>>> df_new
            PHP  Python  Rails  Web-Dev  Beginner  Intermediate
Book Name                                                      
Django 101    0       1      0        1         1             0
ROR Guide     0       0      1        1         0             1
Laravel       1       0      0        1         0             1