if文中で起こる「ValueError: The truth value of a Series is ambiguous.」を解決したい。

Question

前提

Pythonを用いてPyPIを作成しています。

実現したいこと

実際にPyPIでパッケージ化し、pipでインストール、
その後にターミナル上で

$ [パッケージ名] [国名]

といった感じで入力をするとその国の結果をグラフ表示するプログラムを完成させたいです。

発生している問題・エラーメッセージ

Traceback (most recent call last):
  File "/Users/inoseizuru/Desktop/covidill再開発/covidillness.py", line 41, in <module>
    if d.loc[(d.date==j) & (d.location==i),'total_deaths']:
  File "/Users/inoseizuru/opt/anaconda3/lib/python3.9/site-packages/pandas/core/generic.py", line 1527, in __nonzero__
    raise ValueError(
ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

該当のソースコード

import pandas as pd
import subprocess as sp
import sys
import numpy as np
import matplotlib.pyplot as plt
sp.call('wget -nc https://covid.ourworldindata.org/data/owid-covid-data.csv', shell=True)

d=pd.read_csv('owid-covid-data.csv')
d.fillna(0,inplace=True)
lastday=str(d.date.iloc[-1:]).split()[1]
print(lastday)

n=len(sys.argv)-1
print('countries',n)

countries=[]
for i in range(n):
	countries.append(sys.argv[i+1])
# print(countries)

from datetime import date
d0 = date(2020, 2, 29)
d1 = date(int(lastday.split('-')[0]),int(lastday.split('-')[1]),int(lastday.split('-')[2]))
delta = d1 - d0 
days=delta.days

daysdate=sorted(d.date.unique())
daysdate=daysdate[len(daysdate)-days:-1]
# print(daysdate)

dd=pd.DataFrame(
  {
   "date": daysdate,
   "deaths": range(len(daysdate)),
   "icu_patients": range(len(daysdate)),
  })

for i in countries:
	print(i)
	for j in daysdate:
		if d.loc[(d.date==j) & (d.location==i),'total_deaths']:
			dd.loc[dd.date==j,'deaths']=d.loc[(d.date==j) & (d.location==i),'total_deaths']
		dd.loc[dd.date==j,'icu_patients']=d.loc[(d.date==j) & (d.location==i),'icu_patients']
	dd.to_csv(i+'.csv',index=False)

def main():
 for i in range(len(countries)):
  i=pd.read_csv(countries[i]+'.csv') 
  plt.ylabel('ICU')
  plt.plot(i.date,i.icu_patients)
  plt.xticks(np.arange(0,days,30*days/770),rotation=90)
  fig=plt.figure(1)
  fig.set_size_inches(10,3)
 plt.legend(countries)
 plt.savefig('result.png',dpi=fig.dpi,bbox_inches='tight')
 plt.show()

if __name__ == "__main__":
 main()

試したこと

調べたところ、.allなどを使うと解決すると書いてあったので実際に以下のように書いて実行してみたところ、
グラフは表示されましたが、作成されたcsvのデータの中身は空で、グラフは何もプロットされていない真っ白の状態でした。

if all(d.loc[(d.date==j) & (d.location==i),'total_deaths']):

補足情報

色々とググりまくって試行錯誤しながら2日ほど悩んだのですが、
進捗が生まれずといった形です....
もし対処法がわかる方がいらっしゃいましたらご教授願います。

Answer 1

print(type(d.loc[(d.date==j) & (d.location==i),'total_deaths']))

の結果が

<class 'pandas.core.series.Series'>

でSeriesクラスで返ってきており、Bool型やInt型などではないので、if文で評価できないわけです。
TrueかFalseかはっきりしろとのことなので、今回はanyを使いました。allを使ってもいいと思います。

また、

dd.loc[dd.date==j,'deaths']=d.loc[(d.date==j) & (d.location==i),'total_deaths']

も上と同じ様な理由(全く同じというわけではありません)で、

dd.loc[dd.date == j, 'deaths'] = d.loc[(d.date == j) & (d.iso_code == i), 'total_deaths'].values[0]

こうする必要があります。

以下は修正・フォーマットしたコード全文になります。

from datetime import date
import pandas as pd
import subprocess as sp
import sys
import numpy as np
import matplotlib.pyplot as plt

sp.call('wget -nc https://covid.ourworldindata.org/data/owid-covid-data.csv', shell=True)
d = pd.read_csv('owid-covid-data.csv')
d.fillna(0, inplace=True)
lastday = str(d.date.iloc[-1:]).split()[1]
print(lastday)

n = len(sys.argv)-1
print('countries', n)

countries = []
for i in range(n):
    countries.append(sys.argv[i+1])
# print(countries)

d0 = date(2020, 2, 29)
d1 = date(int(lastday.split('-')[0]), int(lastday.split('-')[1]), int(lastday.split('-')[2]))
delta = d1 - d0
days = delta.days

daysdate = sorted(d.date.unique())
daysdate = daysdate[len(daysdate)-days:-1]
# print(daysdate)

dd = pd.DataFrame(
    {
        "date": daysdate,
        "deaths": range(len(daysdate)),
        "icu_patients": range(len(daysdate)),
    })

for i in countries:
    print(i)
    for j in daysdate:
        if d.loc[(d.date == j) & (d.iso_code == i), 'total_deaths'].any():
            dd.loc[dd.date == j, 'deaths'] = d.loc[(d.date == j) & (d.iso_code == i), 'total_deaths'].values[0]
        dd.loc[dd.date == j, 'icu_patients'] = d.loc[(d.date == j) & (d.iso_code == i), 'icu_patients'].values[0]
    dd.to_csv(i+'.csv', index=False)


def main():
    for i in range(len(countries)):
        i = pd.read_csv(countries[i]+'.csv')
        plt.ylabel('ICU')
        plt.plot(i.date, i.icu_patients)
        plt.xticks(np.arange(0, days, 30*days/770), rotation=90)
        fig = plt.figure(1)
        fig.set_size_inches(10, 3)
    plt.legend(countries)
    plt.savefig('result.png', dpi=fig.dpi, bbox_inches='tight')
    plt.show()

if __name__ == "__main__":
    main()

Comments

[i-inose]

詳しい解説ありがとうございます！
実際に変更し、実行してみたところ、このようなエラーが出てしまったのですが、
こちらの解決策、ご教授願います🙇‍♂️

IndexError Traceback (most recent call last)
/var/folders/d2/6n8vvqm55xsgjvv8fl52j9k80000gn/T/ipykernel_11286/667121795.py in <module>
4 if d.loc[(d.date == j) & (d.iso_code == i), 'total_deaths'].any():
5 dd.loc[dd.date == j, 'deaths'] = d.loc[(d.date == j) & (d.iso_code == i), 'total_deaths'].values[0]
----> 6 dd.loc[dd.date == j, 'icu_patients'] = d.loc[(d.date == j) & (d.iso_code == i), 'icu_patients'].values[0]
7 dd.to_csv(i+'.csv', index=False)

IndexError: index 0 is out of bounds for axis 0 with size 0

[i-inose]

続けてすみません。
deathsの方をplotするようにして動かしてみたところ、
グラフ表示はされたのですが、
新たに作られたcsvファイルの中身を見たところdeathsの部分が1,2,3,4,5,6,.......と1ずつ増加しているだけで、
元々のdのデータフレームからコピーされていませんでした....
そのため。グラフも綺麗な一次関数のような一本線となっていました。

[CreeperSaviour]

こちらではJPNで実行しましたが、そちらでは何を入力に与えたかを教えていただけると、こちらでも再現できますのでありがたいです。

[i-inose]

確かにJPNで実行したら動きました！
ですが、locationで指定をしたはずなのでJapanと打ったらできるのかなと思って実行していたのですが、
できなかったという形です....
iso-codeではなく、locationの要素で実行するとグラフが出るようにするにはどうすれば良いでしょうか？？

[CreeperSaviour]

3か所「d.iso_code」となっているところがあるので、それをすべて「d.location」に変えればいいです。

[i-inose]

ありがとうございます！！！
無事解決できました！！