モジュールを使用せずcsv形式のデータを配列に格納した後,特定の要素を持つデータのみを新しい配列に格納する

Question

プログラミング初心者です.

現在,pythonを用いてcsv形式のデータの分析を行っております.
前提として,csv,numpy,pandasなどのモジュールを使用しないプログラムを組んでおります.

全データを配列に格納し,特定の要素を持つデータをすべて抽出し新しい配列に格納したいのですが,プログラムの処理時間が長すぎて結果が表示されないという現象が起こっています.
以下にプログラムを示します.

filepath = 'aaa.csv'


with open(filepath,encoding="shift-jis") as f:
    data2=[]
   
    for line in f:
        line = line.replace('\n','')
        data = line.split(',')
        data2.append(data)
       
i = 0
n = len(data2) 
data3=[]
while i<=n:
    if ((data2[i][0]))=='2018/09/01 00:00':
        data3.append(data2[i])
        i += 1    
print (data3)               
    

処理時間短縮に関してアドバイスよろしくお願いします.

Comments

[tiitoi]

処理時間が長いというよりは while のところで i が n になることがなく、無限ループしているようにみえますがいかがでしょうか。

[TKWTsan]

ご指摘ありがとうございます。while i<nにすることで解決できたのですが、これだと全データ数ー１から抽出することにならないでしょうか。

[tiitoi]

i<=n を i < n にするのではなく、i += 1 が if 文が成立した場合にのみ実行されるというところが問題ではないでしょうか。
data2 の全要素に対して実行したいのであれば、i += 1 のインデントを1つ下げて if 文の外に出す必要があると思います。

[TKWTsan]

全要素に対して実行していることが確認できました。分かりやすい解説ありがとうございました。

Answer 1

自分のページと間違えて投稿してしまいました。失礼しました。

Answer 2

以下のコードで、対象のCSVの一番初めのデータが2018/09/01 00:00だった場合、
そのデータ列を取得することができます。

filepath = 'aaa.csv'

with open(filepath,encoding="shift-jis") as f:
    data2=[]

    for line in f:
        line = line.replace('\n','')
        data = line.split(',')
        data2.append(data)

data3=[]
for i in data2:
    if i[0] == '2018/09/01 00:00':
        data3.append(i)
        
print(data3)

使用したcsvのデータは以下となります。

2018/09/01 00:00, aaa, bbb
2018/09/02 00:00, aaa, bbb
2018/09/01 00:00, ccc, ddd
2018/09/01 01:00, ccc, ddd

実行結果

[['2018/09/01 00:00', ' aaa', ' bbb'], ['2018/09/01 00:00', ' ccc', ' ddd']]