日付を比較演算子で期間指定したい

Question

日付順に並んだCSVファイルがあります。

上記画像の日付部分を期間指定したいのですが、以下のエラーが出ました。

  File "./ameda02.py", line 23
    if date < 2020-06-20 14:17:34:
                    ^
SyntaxError: invalid token

実行したプログラムは下記のものになります。

 cat ameda02.py
#!/usr/bin/env python3

import os
import time
from datetime import datetime
import subprocess
import sys
import csv
import random

def readCsv(ifile='./dat/momo.csv'):

    d_temp = 0.4

    with open(ifile, 'r') as csvfile:
        csv_reader = csv.reader(csvfile, delimiter=',', quotechar='"')
        for row in csv_reader:
            date = row[0]
            temp = float(row[1])
            temp_bak = temp


            if date < 2020-06-20 14:17:34:
                r = 40
            else:
                r = 20

            print('%s,%.1f' %(date, r))

if __name__ == '__main__':
    random.seed()
    ifile = sys.argv[1]
    readCsv(ifile)

調べてもよくわからない状態です。
ご教授いただけると嬉しいです。

よろしくお願い致します。

Comments

[tiitoi]

pandas を使用したやり方でもいいでしょうか？あと CSV の列名がどうなっているか(あるいはヘッダーがないか)を教えて下さい

[momokokomomo]

pandasでも大丈夫です。
列名はないです。
1行目から日付と数値になります。

Answer 1

1列目が日付だと仮定すると、read_csv() で読み込んだあとに df[0] <= 日付を表す文字列 が成り立つ行は40、そうでない行は20を numpy.where() で代入すればよいです。

import numpy as np
import pandas as pd

df = pd.read_csv("sample.csv", header=None, parse_dates=[0])
df[2] = np.where(df[0] <= "2020-06-20T14:17:34", 40, 20)
df.to_csv("output.csv")
print(df)

Comments

[momokokomomo]

ありがとうございます。

上記のプログラムを私のプログラムの
if date < 2020-06-20 14:17:34:
r = 40
else:
r = 20
で用いるためにはどのように書けばいいでしょうか。

[tiitoi]

回答を修正しました。1列目の日付が 2020-06-20 14:17:34 以下の場合は3列目に値を40に設定、そうでない場合は3列目に値20を設定、その後、CSV 形式で修正したものを出力するようにしました

[lehshell]

date は文字列と思いますので
# if date < 2020-06-20 14:17:34:
if date < "2020-06-20 14:17:34":
をお試しください。