[python] pandasで読み込んだcsvファイルの処理について

Question

#pandasで読み込んだcsvデータの取り扱いについて

はじめまして。python初心者です。
先ほども質問したのですが、質問の仕方が悪くご指摘を頂いたので再ポストです。
すみません。

# coding: UTF-8
import pandas as pd
df = pd.read_csv('sample.csv',encoding='shift_jis',usecols=[0,51,52,105,106,109,127])

df[df["109"].str.contains("文字列")]
df.to_csv('aaaa.csv',encoding="shift_jis")

カラムを指定して読み込んでそのカラムの番号内で文字列を検索し、
それ以外の行を削除しようとこころみたのですがうまくいきません。

key errorがでるのですが、どうすればいいかわかりません。。
お手数をおかけしますが宜しくお願い致します。

Comments

[mather]

なんとか埋めようと適当なコードを書いただけですよね…。ぜひ自分自身でpandasの公式リファレンス http://pandas.pydata.org/pandas-docs/stable/index.html を読み解く努力や、他の事例を参考にするなどしてどこまでが理解できていて、どこからがわからないのか具体的に書いてほしいのです。

Answer 1

単にusecolsにて選択された列のなかに "109"という 『column name』 の列が存在しないだけかと思いますが・・。

質問のコードでは usecols に column indexのリストにて 109 を与えておりますが、その列の column name が "109" であるとは限りません(通常はHeader行に記載されている名前になるかと思います)。

とりあえずですが、CSVデータのHeaderがどのように記述されているのかわからないので、column name ではなく column index を使って選択する方法で良いかと思います。

今回の場合、元のcsvデータの 110列目(column_index=109) のデータを使いたいのであれば、読み込み後のデータとしては 6列目(column_index=5) になるので

df[df.iloc[:,5].str.contains("文字列")].to_csv('aaaa.csv', ...)

で良いのではないでしょうか

Comments

[Dadada_c.]

ありがとうございます！そういう意味だったのですね。。助かりました。もっともっと勉強してみます。本当にありがとうございました。