Pandas，结合来自 2 个数据框和一个字典的信息

Question

好吧，这有点难以解释....

假设您有 2 个 pandas 数据框和 1 个字典。

df1 = pd.DataFrame(np.random.randn(5, 3), columns=['b', 'c','d'])
df1['a'] = pd.Series(['1 A1-1','3 A1-1','8 A1-2','17 A1-3','45 A1-16'], index=df1.index)
df1 = df1.reindex_axis(sorted(df1.columns), axis=1)


df2 = pd.DataFrame([['1 A1-1',5],['2 A1-1',8],['3 A1-1',10],['8 A1-2',4],['17 A1-3',1],['45 A1-16',2]], columns = ['m','n'])

dt = {'A1-1':100, 'A1-2':150, 'A1-3':200, 'A1-4':250, 'A1-5':300, 'A1-16':950}

df1['a']和df2['m']包含IDs，有些是相同的。 df2['n'] 包含附加值。 dt 包含 ID 组的基本值，例如 A1-1、A1-2 等

我现在想要 compare/combine df1、df2 和 dt 中的数据，以便我能够向 [=20 添加一个新列=]: 每当 df1['a'] 和 df2['m'] 中的 IDs 相同时，将具有相同字符串部分的字典中的基本值添加到相应的 df2['n'] 中，然后将结果传输到df1['e'].

中的新专栏

我遇到的一个主要问题是 ID 和字典键中的字符串处理：例如 df1 中的 '1 A1-1' 和 [=16 中的 df2 和 'A1-1' =] - 不知道如何比较它们。

最有帮助的结果是 df1['e'] = pd.Series([105,110,154,201,952], index = df1.index)。

感谢您的帮助。

Answer 1

在我看来，这个问题的解释非常好。

第一个split by whitespace column a and select second lists with str[1] and then map by dict and add maped column a by Series created by set_index:

df1['e'] = df1['a'].str.split().str[1].map(dt) + df1['a'].map(df2.set_index('m')['n'])
print (df1)
          a         b         c         d    e
0    1 A1-1  0.026375  0.260322 -0.395146  105
1    3 A1-1 -0.204301 -1.271633 -2.596879  110
2    8 A1-2  0.289681 -0.873305  0.394073  154
3   17 A1-3  0.935106 -0.015685  0.259596  201
4  45 A1-16 -1.473314  0.801927 -1.750752  952

编辑：

map 函数使用 dict 的键替换某些列中的值。类似系列地图，仅使用 instaed 键 index 值，而不是 values 使用值。

#sample data
df = pd.DataFrame({'a':['bar','foo', 'baz'], 'b':[7,8,9]})
print (df)
     a  b
0  bar  7
1  foo  8
2  baz  9

#dict and df for mapping
d = {'foo':15, 'bar':20}
df2 = pd.DataFrame({'m':['baz','bar','foo'], 'n':[3,4,5]})
print (df2)
     m  n
0  baz  3
1  bar  4
2  foo  5

#create Series for map
print (df2.set_index('m')['n'])
m
baz    3
bar    4
foo    5
Name: n, dtype: int64

df['c'] = df['a'].map(d)
df['d'] = df['a'].map(df2.set_index('m')['n'])
print (df)
     a  b     c  d
0  bar  7  20.0  4
1  foo  8  15.0  5
2  baz  9   NaN  3

Pandas，结合来自 2 个数据框和一个字典的信息

Pandas, combining information from 2 dataframes and one dictionary

python

dictionary

string-comparison

dataframe

pandas