将特定值移动到相同的值列中

Question

我试图将这些值分开，并用数值将它们排序到另一个df中。如果ID中出现任何值，则在新df中显示为1，如果该值不存在，则显示为0。

ID   Values
0    1,2,3  
1    2,5,7,8
2    4,5,10

Results

ID 1 2 3 4 5 6 7 8 9 10
0  1 1 1 0 0 0 0 0 0 0
1  0 1 0 0 1 0 1 1 0 0
2  0 0 0 1 1 0 0 0 0 1

Answer 1

我们可以使用eval将Values列转换为元组格式，然后使用get\u dummies方法获得预期结果：

>>> df['Values'] = df['Values'].apply(eval)
>>> pd.get_dummies(df['Values'].apply(pd.Series).stack().astype(int)).sum(level=0)
    1   2   3   4   5   7   8   10
0   1   1   1   0   0   0   0   0
1   0   1   0   0   1   1   1   0
2   0   0   0   1   1   0   0   1

Answer 2

在get_dummies之后，我们仍然需要重新索引以找回丢失的列

s = df.join(df.pop('Values').str.get_dummies(',')).set_index('ID')
s.columns = s.columns.astype(int)
s = s.reindex(columns = range(1, s.columns.max()+1),fill_value=0)
s
Out[61]: 
    1   2   3   4   5   6   7   8   9   10
ID                                        
0    1   1   1   0   0   0   0   0   0   0
1    0   1   0   0   1   0   1   1   0   0
2    0   0   0   1   1   0   0   0   0   1