pandas中常用的一件事情就是对特定条件进行搜索,那么这里介绍使用pandas搜索方式,本案例使用的pandas是anaconda中的,可以下载任意的anaconda进行安装都会带有。
首先导入包
1
2
|
import pandas as pd
import os
|
然后需要构建一个csv文件存上文件。比如我们有一个叫test.csv的文件,文件有三个列的表头,id、name、address我们想知道一个人的名字的id与地址就可以写成
1
2
|
test_csv = pd.read_csv( '/test.csv' , encoding = 'gb2312' ) # 如果有中文就要gb2312
info_Joan = test_csv[test_csv[ 'name' ] = 'Join] # 这样我们就获得了Joan的id以及address 信息
|
或者是
1
|
info_Joan = test_csv[test_csv[ 'name' ]isin([ 'Join' ])]
|
余下的就是你想要的操作了。比如取得values的或者组成列表
1
|
[ list (x) for x in info_Joan[ 'id' ].values]
|
以上这篇pandas 条件搜索返回列表的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持服务器之家。
原文链接:https://blog.csdn.net/benofdog/article/details/72588209