сводный кадр данных с повторяющимися значениями - PullRequest
1 голос
/ 19 июня 2019

рассмотрите ниже pd.DataFrame

temp = pd.DataFrame({'label_0':[1,1,1,2,2,2],'label_1':['a','b','c',np.nan,'c','b'], 'values':[0,2,4,np.nan,8,5]})

print(temp)
        label_0 label_1 values
    0   1           a   0.0
    1   1           b   2.0
    2   1           c   4.0
    3   2          NaN  NaN
    4   2           c   8.0
    5   2           b   5.0

мой желаемый результат -

     label_1    1   2
  0     a      0.0  NaN
  1     b      2.0  5.0
  2     c      4.0  8.0
  3     NaN    NaN  NaN

Я пробовал pd.pivot и спорю с pd.gropuby, но не могу добраться до желаемоговывод из-за повторяющихся записей.любая помощь наиболее ценится.

Ответы [ 4 ]

3 голосов
/ 19 июня 2019

Другой способ - использовать set_index и unstack:

temp.set_index(['label_0','label_1'])['values'].unstack(0)

Выход:

label_0    1    2
label_1          
NaN      NaN  NaN
a        0.0  NaN
b        2.0  5.0
c        4.0  8.0
3 голосов
/ 19 июня 2019
d = {}
for _0, _1, v in zip(*map(temp.get, temp)):
    d.setdefault(_1, {})[_0] = v

pd.DataFrame.from_dict(d, orient='index')

       1    2
a    0.0  NaN
b    2.0  5.0
c    4.0  8.0
NaN  NaN  NaN

ИЛИ

pd.DataFrame.from_dict(d, orient='index').rename_axis('label_1').reset_index()

  label_1    1    2
0       a  0.0  NaN
1       b  2.0  5.0
2       c  4.0  8.0
3     NaN  NaN  NaN
2 голосов
/ 19 июня 2019

Вы можете сделать fillna, затем pivot

temp.fillna('NaN').pivot(*temp.columns).T
Out[251]: 
label_0    1    2
label_1          
NaN      NaN  NaN
a          0  NaN
b          2    5
c          4    8
0 голосов
/ 19 июня 2019

Похоже на простое pivot работает:

temp.pivot(columns='label_0', index='label_1', values='values')

Вывод:

label_0     1       2
label_1         
NaN         NaN     NaN
a           0.0     NaN
b           2.0     5.0
c           4.0     8.0
...