pandas 条件搜索返回列表的方法
pandas中常用的一件事情就是对特定条件进行搜索,那么这里介绍使用pandas搜索方式,本案例使用的pandas是anaconda中的,可以下载任意的anaconda进行安装都会带有。
首先导入包
import pandas as pd import 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]
以上这篇pandas 条件搜索返回列表的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持【听图阁-专注于Python设计】。