ifで分岐させた異なる計算を同一の変数に入れたい

Question

前提・実現したいこと

meshgridで作成したlist_thetaの0行目と1行目で計算を行うのですが、値の大きさによって計算内容を変え、それを同一のtheta_ijに入れたいのですが、一番最後の行で以下のエラーが発生しました。

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

 values[indexer] = value

ValueError: setting an array element with a sequence.

該当のソースコード

import pandas as pd
import numpy as np

df_read_North=pd.read_csv("Uraga_GateCrossingList_20180101NS.csv",encoding="shift-jis")
df_read_South=pd.read_csv("Uraga_GateCrossingList_20180101NS.csv",encoding="shift-jis")
df_read_West=pd.read_csv("Uraga_GateCrossingList_20180101WE.csv",encoding="shift-jis")
df_read_East=pd.read_csv("Uraga_GateCrossingList_20180101WE.csv",encoding="shift-jis")

df_data_North=df_read_North[(df_read_North["i"]==0) & (df_read_North["j"]==0) & (df_read_North["SogKnot"]!=0)]
df_data_South=df_read_South[(df_read_South["i"]==1) & (df_read_South["j"]==0) & (df_read_South["SogKnot"]!=0)]
df_data_West=df_read_West[(df_read_West["i"]==0) & (df_read_West["j"]==0) & (df_read_West["SogKnot"]!=0)]
df_data_East=df_read_East[(df_read_East["i"]==1) & (df_read_East["j"]==0) & (df_read_East["SogKnot"]!=0)]

df_data_NS=pd.merge(df_data_North,df_data_South,how="outer")
df_data_WE=pd.merge(df_data_West,df_data_East,how="outer")
df_data_merge=pd.merge(df_data_NS,df_data_WE,how="outer")

df_theta=df_data_merge.groupby('Group').mean()['θ']

list_theta=np.meshgrid(df_theta,df_theta)

NN=len(df_theta)
theta_ij=np.abs(list_theta[0]-list_theta[1])
df_theta_ij=pd.DataFrame(theta_ij)

for a in range(NN):
    for b in range(NN):
        if df_theta_ij.iloc[a,b]<180:
            df_theta_ij.iloc[a,b]=360-np.abs(list_theta[0]-list_theta[1])

試したこと

行番号と列番号を繰り返し指定して、条件に合うときだけ、値を更新したいので、データフレームにしないでやってみたりしたのですが、ダメでした。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Answer 1

if np.abs(list_theta[0]-list_theta[1])<180:

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

「ValueError：複数の要素を持つ配列の真理値はあいまいです。 a.any（）またはa.all（）を使用して下さい」と言われていますね。

多文if文で怒られているのだと思います。このようにしっかりと判定すれば行けませんか？

if len(list_theta) >= 2:
    if np.abs(list_theta[0] - list_theta[1])<180:
        theta_ij = np.abs(list_theta[0] - list_theta[1])
    else:
        theta_ij = 360 - np.abs(list_theta[0] - list_theta[1])
else:
    theta_ij = 0

Comments

[ryosuke197]

回答ありがとうございます。やってみましたが、同じくif np.abs～<180:の所で同じエラーが発生してしまいました。そして、if文の条件式は、1つの値に対してなのかなと思い、行と列を繰り返し指定して、df_theta_ijを更新するように書き換えてみました。これもエラーが出たのですが、どこがおかしいか確認していただけませんか。

[ryosuke197]

すみません、解決しました。