如何将DataFrame的genres列拆分为对应20个流派的独立列?
解决方法
你需要的是独热编码(One-Hot Encoding),而非简单的字符串拆分——后者只能按每行的流派数量生成对应列,无法固定映射到20个指定流派。直接用以下方法即可实现需求:
一步生成所有流派列
使用pandas内置的str.get_dummies方法,自动识别所有唯一流派并生成对应标记列:
import pandas as pd # 生成独热编码列,列名就是流派名称,值为1(包含该流派)/0(不包含) genre_columns = df['genres'].str.get_dummies(sep='|')
合并到原DataFrame
如果需要把这些新列和原数据合并:
df = pd.concat([df, genre_columns], axis=1)
为什么原代码失效?
str.split(expand=True)只是将每行的流派按分隔符拆分成若干列,列数由该行最多的流派数量决定(你的数据里最多5个),且这些列没有固定对应某个流派,完全无法满足“每个流派对应固定列”的需求。
自定义指定20个流派
如果数据中流派超过20种,你可以先指定目标流派列表,再筛选对应列:
# 假设你的20个目标流派列表 target_genres = ['Action', 'Adventure', 'Horror', 'Comedy', ...] # 补全剩余流派 # 只保留目标流派的列 genre_columns = genre_columns[target_genres]
内容的提问来源于stack exchange,提问作者possizola
相关产品推荐
相关产品推荐

