Python 条件にあったものの抽出

Question

PythonでCSVのデータを読み込み、条件にあったデータを取りたいです。

1900/01/01,yamada,40
1900/01/02,tanaka,30
1900/01/03,hanako,100
1900/01/04,aoki,90
1900/01/05,tanaka,89
1900/01/05,yamada,30

というデータがあり左から日時・名前・点数を表しています。
この時、scoreが90点以上の人を抽出したいです。

抽出する際は ['1900/01/04','aoki','90'] の様になることを想定しています。

試してみたこと

import csv

file = 'test.csv'
with open(file,'r') as f:
    rows = csv.reader(f)
    for row in rows:
        print(row[3] >= 90)

Comments

[TakaiY]

提示されたコードだとどのような問題があるのですか？

[退会済みユーザー]

最初はtrue か falseか表示されていたのですが、エラーも出ずにprintされません

Answer 1

こんにちは

以下でどうでしょう？

import pandas as pd

df = pd.read_csv('test.csv', names=('date', 'student_name', 'score'))

ls = [[*row] for _, row in df[df.score >= 90].iterrows()]

print(ls)  # => [['1900/01/03', 'hanako', 100], ['1900/01/04', 'aoki', 90]]

• 動作確認用Repl.it: https://repl.it/@jun68ykt/Q290228

Answer 2

以下の部分に3つ問題があります。

print(row[3] >= 90)

1. row[3]だと、4番目の要素を示しています。
2. row[2}だとしても、これは文字列ですので、 数値である90とは比較できません。90のような数値と比較したいのでされば、 int(row[2])などとして数値化する必要があります。
3. row[3] >= 90 そのもは比較の結果ですから、質問の回答にあるようにTrueかFalseがprintされます。

なので、最後の部分は、

if int(row[2]) >= 90:
    print (row)

などとすればいいのではないかと思います。

Answer 3

• if文で選択
• indexが3になっていたので2に修正
• 比較のためにint型

と修正しました。

import csv

file = 'test.csv'
with open(file,'r') as f:
    rows = csv.reader(f)
    for row in rows:
        if int(row[2]) >= 90:
            print(row)
            
#['1900/01/03', 'hanako', '100']
#['1900/01/04', 'aoki', '90']