1

私は次の辞書を持っています：

d = {'col1': ['a', 'b', 'c'],
     'col2': [[1,2], [4,3,2], []],
}

次のような Pandas DataFrame が必要です。

idx, col1, col2
  0,  'a',  1
  1,  'a',  2
  2,  'b',  4
  3,  'b',  3
  4,  'b',  2
  5   'c',  nan

これはどのように構築できますか？dict を渡すだけでは、col2 のリスト項目が解明/繰り返されません。ありがとう！

4

1 に答える 1

3

自分で構築するしかありません。1 つの方法を次に示します。

col1 = ['a', 'b', 'c']
col2 = [[1,2], [4,3,2], []]
col2_lens = map(len, col2)

# flatten col2
s2 = Series([eli for el in col2 for eli in (el or [nan])])

# replicate elements of col1 col2[i] times
s1 = Series(list(''.join(el * (col2_len or 1) for el, col2_len in zip(col1, col2_lens))))
concat([s1, s2], axis=1)

降伏

   0   1
0  a   1
1  a   2
2  b   4
3  b   3
4  b   2
5  c NaN

ここに%%timeit示されている 3 つのメソッドの s を次に示します。

1.

%%timeit
col2_lens = map(len, col2)

# flatten col2
s2 = Series([eli for el in col2 for eli in (el or [nan])])

# replicate elements of col1 col2[i] times
s1 = Series(list(''.join(el * (col2_len or 1) for el, col2_len in zip(col1, col2_lens))))
concat([s1, s2], axis=1)

1000 loops, best of 3: 646 µs per loop

2.

%%timeit
df = DataFrame()
for a, b in zip(col1, col2):
    df = pd.concat([df, pd.DataFrame({'col1': a, 'col2': b or [np.nan]})])

100 loops, best of 3: 2.52 ms per loop

3.

%%timeit
frames = []
for a, b in zip(col1, col2):
    frames.append(pd.DataFrame({'col1': a, 'col2': b or [np.nan]}))
df = pd.concat(frames)

1000 loops, best of 3: 1.58 ms per loop
于 2013-08-16T21:50:43.513 に答える