pandas 条件搜索返回列表

来源:互联网 发布:软件添加广告sdk 编辑:程序博客网 时间:2024/05/29 19:56

pandas中常用的一件事情就是对特定条件进行搜索,那么这里介绍使用pandas搜索方式,本案例使用的pandas是anaconda中的,可以下载任意的anaconda进行安装都会带有。

首先导入包

import pandas as pdimport os
然后需要构建一个csv文件存上文件。比如我们有一个叫test.csv的文件,文件有三个列的表头,id、name、address我们想知道一个人的名字的id与地址就可以写成
test_csv = pd.read_csv('/test.csv', encoding='gb2312')# 如果有中文就要gb2312
info_Joan = test_csv[test_csv['name']='Join] # 这样我们就获得了Joan的id以及address 信息
或者是
info_Joan =test_csv[test_csv['name']isin(['Join'])]
余下的就是你想要的操作了。比如取得values的或者组成列表
[list(x) for x in info_Joan['id'].values]




原创粉丝点击