怎样才能最有效地组织下列熊猫数据框架:
data =
Position Letter
1 a
2 b
3 c
4 d
5 e
变成一个像字母一样的字典[1:'a', 2: 'b', 3: 'c', 4: 'd', 5: 'e']?
怎样才能最有效地组织下列熊猫数据框架:
data =
Position Letter
1 a
2 b
3 c
4 d
5 e
变成一个像字母一样的字典[1:'a', 2: 'b', 3: 'c', 4: 'd', 5: 'e']?
当前回答
在Python 3.6中,最快的方法仍然是WouterOvermeire。Kikohs的提议比其他两个方案要慢。
import timeit
setup = '''
import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.randint(32, 120, 100000).reshape(50000,2),columns=list('AB'))
df['A'] = df['A'].apply(chr)
'''
timeit.Timer('dict(zip(df.A,df.B))', setup=setup).repeat(7,500)
timeit.Timer('pd.Series(df.A.values,index=df.B).to_dict()', setup=setup).repeat(7,500)
timeit.Timer('df.set_index("A").to_dict()["B"]', setup=setup).repeat(7,500)
结果:
1.1214002349999777 s # WouterOvermeire
1.1922008498571748 s # Jeff
1.7034366211428602 s # Kikohs
其他回答
我找到了一个更快的方法来解决这个问题,至少在实际的大型数据集上使用: df.set_index(关键).to_dict()(价值)
5万行证明:
df = pd.DataFrame(np.random.randint(32, 120, 100000).reshape(50000,2),columns=list('AB'))
df['A'] = df['A'].apply(chr)
%timeit dict(zip(df.A,df.B))
%timeit pd.Series(df.A.values,index=df.B).to_dict()
%timeit df.set_index('A').to_dict()['B']
输出:
100 loops, best of 3: 7.04 ms per loop # WouterOvermeire
100 loops, best of 3: 9.83 ms per loop # Jeff
100 loops, best of 3: 4.28 ms per loop # Kikohs (me)
下面是用以下df测试的另外两种方法。
df = pd.DataFrame(np.random.randint(0,10,10000).reshape(5000,2),columns=list('AB'))
使用to_records ()
dict(df.to_records(index=False))
使用MultiIndex.from_frame ()
dict(pd.MultiIndex.from_frame(df))
每个时间。
24.6 ms ± 847 µs per loop (mean ± std. dev. of 7 runs, 10 loops each)
1.86 ms ± 11.6 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)
我喜欢Wouter方法,但是具有重复值的行为可能不是预期的,不幸的是,OP没有以一种方式或其他方式讨论这种情况。Wouter,将始终为遇到的每个键选择最后一个值。换句话说,它会一直覆盖每个键的值。
在我的脑海中,预期的行为更像是使用dataframe中的两列创建一个字典,其中一列中的副本为每个键保留一个列表。
因此,对于保持副本的情况,让我提交df.groupby('Position')['Letter'].apply(list).to_dict()(甚至可能是一个集合而不是列表)
In [9]: pd.Series(df.Letter.values,index=df.Position).to_dict()
Out[9]: {1: 'a', 2: 'b', 3: 'c', 4: 'd', 5: 'e'}
速度比较(使用Wouter方法)
In [6]: df = pd.DataFrame(randint(0,10,10000).reshape(5000,2),columns=list('AB'))
In [7]: %timeit dict(zip(df.A,df.B))
1000 loops, best of 3: 1.27 ms per loop
In [8]: %timeit pd.Series(df.A.values,index=df.B).to_dict()
1000 loops, best of 3: 987 us per loop
在Python 3.6中,最快的方法仍然是WouterOvermeire。Kikohs的提议比其他两个方案要慢。
import timeit
setup = '''
import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.randint(32, 120, 100000).reshape(50000,2),columns=list('AB'))
df['A'] = df['A'].apply(chr)
'''
timeit.Timer('dict(zip(df.A,df.B))', setup=setup).repeat(7,500)
timeit.Timer('pd.Series(df.A.values,index=df.B).to_dict()', setup=setup).repeat(7,500)
timeit.Timer('df.set_index("A").to_dict()["B"]', setup=setup).repeat(7,500)
结果:
1.1214002349999777 s # WouterOvermeire
1.1922008498571748 s # Jeff
1.7034366211428602 s # Kikohs