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')# 如果有中文就要gb2312info_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 條件搜索返回列表的方法就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持武林站長站。
|
新聞熱點
疑難解答