python - 如何使用其中一列作为引用来匹配两列？

Question

我做了一些分析并发现了一个特定的模式，现在我正在尝试做一些预测。我有一个数据集，可以预测童年时期发生一定数量事故的学生的评分。我的预测矩阵看起来像这样:

   A
   injuries      ratings  
         0            5
         1            4.89
         2            4.34
         3            3.99 
         4            3.89
         5            3.77 

我的数据集如下所示:

B

siblings income injuries total_scoldings_from father
     3       12000   4             09
     4       34000   5             22
     1       23400   3             12
     3       24330   1              1
     0       12000   1             12 

现在我想创建一个列名预测，它基本上与从A到B的条目匹配并返回

siblings income injuries total_scoldings_from_father predictions
     3       12000   4             09                    3.89
     4       34000   5             22                    3.77
     1       23400   3             12                    3.99
     3       24330   1             1                     4.89
     0       12000   1             12                    4.89

请帮忙

还建议一个标题，因为我的标题缺乏供将来引用的所有重要内容

Answer 1

您可以使用map如果映射的所有值都在 DataFrame A 中:

B['predictions'] = B['injuries'].map(A.set_index('injuries')['ratings'])
print (B)
   siblings  income  injuries  total_scoldings_from_father  predictions
0         3   12000         4                            9         3.89
1         4   34000         5                           22         3.77
2         1   23400         3                           12         3.99
3         3   24330         1                            1         4.89
4         0   12000         1                           12         4.89

另一个解决方案 merge :

C = pd.merge(B,A)
print (C)
   siblings  income  injuries  total_scoldings_from_father  ratings
0         3   12000         4                            9     3.89
1         4   34000         5                           22     3.77
2         1   23400         3                           12     3.99
3         3   24330         1                            1     4.89
4         0   12000         1                           12     4.89