Question

我有一系列的熊猫键，想通过从其他数据框中选择值来创建一个数据框。

例如

data_df = pandas.DataFrame({'key' : ['a','b','c','d','e','f'],
                       'value1': [1.1,2,3,4,5,6],
                       'value2': [7.1,8,9,10,11,12]
})
keys = pandas.Series(['a','b','a','c','e','f','a','b','c'])

data_df
#   key value1  value2
#0  a   1.1     7.1
#1  b   2.0     8.0
#2  c   3.0     9.0
#3  d   4.0     10.0
#4  e   5.0     11.0
#5  f   6.0     12.0

我想得到这样的结果

result
    key value1  value2
0   a   1.1     7.1
1   b   2.0     8.0
2   a   1.1     7.1
3   c   3.0     9.0
4   e   5.0     11.0
5   f   6.0     12.0
6   a   1.1     7.1
7   b   2.0     8.0
8   c   3.0     9.0

我成功完成此操作的一种方法是使用

def append_to_series(key):
    new_series=data_df[data_df['key']==key].iloc[0]
    return new_series
pd.DataFrame(key_df.apply(append_to_series))

但是，此功能非常慢且不干净。有办法更有效地做到这一点吗？

Answer 1

通过key列创建索引，然后使用DataFrame.reindex或DataFrame.loc：

通知：原始key列的必要唯一值。

df = data_df.set_index('key').reindex(keys.rename('key')).reset_index()

或者：

df = data_df.set_index('key').loc[keys].reset_index()

print (df)
  key  value1  value2
0   a     1.1     7.1
1   b     2.0     8.0
2   a     1.1     7.1
3   c     3.0     9.0
4   e     5.0    11.0
5   f     6.0    12.0
6   a     1.1     7.1
7   b     2.0     8.0
8   c     3.0     9.0

Answer 2

将系列转换为列名为dataframe的{{1}}

使用pd.merge()合并key

value1,value2

keys = pd.DataFrame(['a','b','a','c','e','f','a','b','c'],columns=['key'])
res = pd.merge(keys,data_df,on=['key'],how='left')

print(res)

根据一系列键和key：value数据框构造数据框

2 个答案: