[英]Pandas : How to split a column into multiple columns by separator?
我有一个数据框:
df ={'date' : ['2020-08-05', '2020-08-05', '2020-08-05'], 'values_a':['jbl_1;jbl2', 'jbl44;jbl441;imax76;wer43', 'macbook12;iphone43;micromax12;ios11'], 'types' : ['connector1','connector1','connector1'], 'connection' : ['working','working','working']}
df = pd.DataFrame(df)
date values_a types connection
0 2020-08-05 jbl_1;jbl2 connector1 working
1 2020-08-05 jbl44;jbl441;imax76;wer43 connector1 working
2 2020-08-05 macbook12;iphone43;micromax12;ios11 connector1 working
我在找什么:
我想使用分隔符拆分 values_a 列并制作额外的列:
我试过的:
def generate_one_hot(df):
# make a list of all unique columns values
all_columns = reduce(operator.concat,[column.split(';') for column in df['values_a']])
# fill one hot values
all_values = [[1 if column_name in value else 0 for value in df['values_a']] for column_name in all_columns]
# map it
dataframe = pd.DataFrame({col_name:col_val for col_name, col_val in zip(all_columns,all_values)})
return pd.concat([df,dataframe],1)
哪个在做这项工作,如何使用本机 pandas 函数优化此代码?
我使用get_dummies
尝试了这个,它正在解决:
one_hot = pd.concat([df,df.values_a.str.get_dummies(sep=';')],1)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.