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]

相关推荐