您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

python – 根据两列的值选择pandas帧行

5b51 2022/1/14 8:23:02 python 字数 1908 阅读 587 来源 www.jb51.cc/python

我希望根据两个列值选择一些特定的行.例如: d = {'user' : [1., 2., 3., 4] ,'item' : [5., 6., 7., 8.],'f1' : [9., 16., 17., 18.], 'f2':[4,5,6,5], 'f3':[4,5,5,8]} df = pd.DataFrame(d) print df Out: f1 f2 f3 item user

概述

d = {'user' : [1.,2.,3.,4],'item' : [5.,6.,7.,8.],'f1' : [9.,16.,17.,18.],'f2':[4,5,6,5],'f3':[4,8]}
df = pd.DataFrame(d)
print df

Out:
   f1  f2  f3  item  user
0   9   4   4     5     1
1  16   5   5     6     2
2  17   6   5     7     3
3  18   5   8     8     4

我想根据’user’和’item’的值选择行.给定一个存储[user,item]值对的2d numpy数组:

samples = np.array([[1,[3,7],[2,6]]) 
Out: 
array([[1,6]])

然后预期的输出是:

Out:
   f1  f2  f3  item  user
0   9   4   4     5     1
2  17   6   5     7     3
2  17   6   5     7     3
1  16   5   5     6     2

然后,我的最终目标是获得一个2d numpy数组存储除item和user之外的所有列值,即:

Out: 
array([[9,4,[17,[16,5]])

我们可以看到,它是列f1,f2,f3的值.

我怎样才能做到这一点?

result = pd.merge(samples,df,how='inner')

产量

user  item  f1  f2  f3
0     1     5   9   4   4
1     3     7  17   6   5
2     3     7  17   6   5
3     2     6  16   5   5
import numpy as np
import pandas as pd

d = {'user' : [1.,8]}
df = pd.DataFrame(d)
samples = np.array([[1,6]]) 
samples = pd.DataFrame(samples,columns=['user','item'])

result = pd.merge(samples,how='inner')
result = result[['f1','f2','f3']]
result = result.values
print(result)

产量

[[  9.   4.   4.]
 [ 17.   6.   5.]
 [ 17.   6.   5.]
 [ 16.   5.   5.]]

总结

以上是编程之家为你收集整理的python – 根据两列的值选择pandas帧行全部内容,希望文章能够帮你解决python – 根据两列的值选择pandas帧行所遇到的程序开发问题。


如果您也喜欢它,动动您的小指点个赞吧

除非注明,文章均由 laddyq.com 整理发布,欢迎转载。

转载请注明:
链接:http://laddyq.com
来源:laddyq.com
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。


联系我
置顶