実現したいこと
openposeのjsonファイルをCSVファイルに変換したいです。
下記の写真のような空のjsonファイルを無視してCSVファイルに変換する方法を教えて頂きたいです。
例外処理をしない状態だと下記のエラーがでます。
下記のコードのように例外処理をtryで行いたかったのですが、時間がかかる&実行できない状態です。
発生している問題・エラーメッセージ
--------------------------------------------------------------------------- IndexError Traceback (most recent call last) ~\AppData\Local\Temp/ipykernel_28972/3865491861.py in <module> 40 41 if __name__ == '__main__': ---> 42 main() ~\AppData\Local\Temp/ipykernel_28972/3865491861.py in main() 37 writer.writerow(["Nose", "Neck", "RShoulder", "RElbow", "RWrist", "LShoulder", "LElbow", "LWrist", "MidHip", "RHip", \ 38 "RKnee", "RAnkle", "LHip", "LKnee", "LAnkle", "REye", "LEye", "REar", "LEar", "LBigToe", "LSmallToe", "LHeel", "RBigToe", "RSmallToe", "RHeel"]) ---> 39 getSpecificData(filelist) 40 41 if __name__ == '__main__': ~\AppData\Local\Temp/ipykernel_28972/3865491861.py in getSpecificData(filelist) 13 with open(filelist[i]) as f: 14 data = json.load(f) ---> 15 data = np.array(data['people'][0]['pose_keypoints_2d']).reshape(-1,3) 16 df = pd.DataFrame(data, columns=['X','Y','P'], index=["Nose", "Neck", "RShoulder", "RElbow", "RWrist", "LShoulder", "LElbow", "LWrist", "MidHip", "RHip", \ 17 "RKnee", "RAnkle", "LHip", "LKnee", "LAnkle", "REye", "LEye", "REar", "LEar", "LBigToe", "LSmallToe", "LHeel", "RBigToe", "RSmallToe", "RHeel"]) IndexError: list index out of range
該当のソースコード
python
1import json 2import pandas as pd 3import numpy as np 4import glob 5import csv 6 7def getFileName(path): 8 filelist = glob.glob(path+ "/*.json") 9 return filelist 10 11def getSpecificData(filelist): 12 for i in range(len(filelist)): 13 while True: 14 try: 15 16 with open(filelist[i]) as f: 17 data = json.load(f) 18 data = np.array(data['people'][0]['pose_keypoints_2d']).reshape(-1,3) 19 df = pd.DataFrame(data, columns=['X','Y','P'], index=["Nose", "Neck", "RShoulder", "RElbow", "RWrist", "LShoulder", "LElbow", "LWrist", "MidHip", "RHip", \ 20 "RKnee", "RAnkle", "LHip", "LKnee", "LAnkle", "REye", "LEye", "REar", "LEar", "LBigToe", "LSmallToe", "LHeel", "RBigToe", "RSmallToe", "RHeel"]) 21 22 # 自分の必要なデータを取り出す 23 writeCSV([float(df.at["Nose", "X"]),float(df.at["Nose", "Y"]),float(df.at["Neck", "X"]),float(df.at["Neck", "Y"]),float(df.at["RShoulder", "X"]),float(df.at["RShoulder", "Y"]),float(df.at["RElbow", "X"]),float(df.at["RElbow", "Y"]), float(df.at["RWrist", "X"]), float(df.at["RWrist", "Y"]) ,float(df.at["LShoulder", "X"]),float(df.at["LShoulder", "Y"]),float(df.at["LElbow", "X"]) \ 24 , float(df.at["LElbow", "Y"]), float(df.at["LWrist", "X"]), float(df.at["LWrist", "Y"]), float(df.at["MidHip", "X"]), float(df.at["MidHip", "Y"]), float(df.at["LWrist", "X"]), float(df.at["LWrist", "Y"]), float(df.at["RHip", "X"]), float(df.at["RHip", "Y"]), float(df.at["RKnee", "X"]), float(df.at["RKnee", "Y"]), float(df.at["RAnkle", "X"]), float(df.at["RAnkle", "Y"]), float(df.at["LHip", "X"]), float(df.at["LHip", "Y"])\ 25 , float(df.at["LKnee", "X"]), float(df.at["LKnee", "Y"]), float(df.at["LAnkle", "X"]), float(df.at["LAnkle", "Y"]), float(df.at["REye", "X"]), float(df.at["REye", "Y"]), float(df.at["LEye", "X"]), float(df.at["LEye", "Y"]), float(df.at["REar", "X"]), float(df.at["REar", "Y"]), float(df.at["LEar", "X"]), float(df.at["LEar", "Y"]), float(df.at["LBigToe", "X"]), float(df.at["LBigToe", "Y"])\ 26 , float(df.at["LSmallToe", "X"]), float(df.at["LSmallToe", "Y"]), float(df.at["LHeel", "X"]), float(df.at["LHeel", "Y"]), float(df.at["RBigToe", "X"]), float(df.at["RBigToe", "Y"]), float(df.at["RSmallToe", "X"]), float(df.at["RSmallToe", "Y"]), float(df.at["RHeel", "X"]), float(df.at["RHeel", "Y"])]) 27 28 except IndexError: 29 pass 30 31 32def writeCSV(data): 33 with open('openposetameshi.csv', 'a') as f: #変更 34 writer = csv.writer(f, lineterminator='\n') 35 writer.writerow(data) 36 37def main(): 38 filelist = getFileName(input("JSONのディレクトリのパスを入力: ")) 39 with open('openposetameshi.csv', 'w') as f: #変更 40 writer = csv.writer(f, lineterminator='\n') 41 # 自分の必要なデータの列の名前を用意。上のデータと同じだけの列数を揃える。 42 writer.writerow(["Nose", "Neck", "RShoulder", "RElbow", "RWrist", "LShoulder", "LElbow", "LWrist", "MidHip", "RHip", \ 43 "RKnee", "RAnkle", "LHip", "LKnee", "LAnkle", "REye", "LEye", "REar", "LEar", "LBigToe", "LSmallToe", "LHeel", "RBigToe", "RSmallToe", "RHeel"]) 44 getSpecificData(filelist) 45 46if __name__ == '__main__': 47 main()
試したこと
ここに問題に対して試したことを記載してください。
補足情報(FW/ツールのバージョンなど)
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。