Python　リスト　指定した行、列ごとに出力させてCSVファイルを整形したい

前提・実現したいこと

実際に用いるCSVファイルは大きすぎるため、簡略化させていただきます。
以下のような5×7のa.csvファイルから任意の行を抜き取ってCSVファイル(out.csv)を整形したいです。

a.csv=[[a,b,c,1,d,e,f,g],[a,b,c,0,h,i,j,k],[a,b,c,1,l,m,n,o],[a,b,c,1,p,q,r,s],[a,b,c,0,t,u,v,w]]

抜き出すルールは次のようになります。

各行の３列目の数字に着目し、0ならばその行の4、5列目の要素を取り出す。まず3列目が0のときの操作を一通り行う。次に1ならばその行の6、7列目の要素を取り出す。

最終形
out.csv=[[1,d,e],[1,l,m],[1,p,q],[0,j,k],[9,v,w]]

上記のような5×3のCSVに整形したいです。

試したこと

単純に4、5列目だけを取り出して並べる、ということはできました。しかし、これでは３列目に指定された数字を全く着目していません。forやifを使ってうまく分岐させたりしたいのですがうまくいきません。

補足情報

実際にはa.csvのファイルは100×100のような大きなものであり、1つずつ入力していくのは難しいのでforやifを使いながら整理して書いていきたいです。
個人的にリストが使いやすそうということでリストで整形を行おうとしましたが、リストで実装することができないようであればnampyなどでも構いません。
補足点等ありましたらコメントいただけると幸いです。

行動規範の内容に同意します

回答1件

ベストアンサー

リストでやるなら以下でできています。

python
1>>> a_csv = [['a','b','c','1','d','e','f','g'],['a','b','c','0','h','i','j','k'],['a','b','c','1','l','m','n','o'],['a','b','c','1','p','q','r','s'],['a','b','c','0','t','u','v','w']]
2>>> out_csv = [row[3:4]+row[4:6] for row in a_csv if row[3] == '1'] + [row[3:4]+row[6:8] for row in a_csv if row[3] == '0']
3>>> print(out_csv)
4[['1', 'd', 'e'], ['1', 'l', 'm'], ['1', 'p', 'q'], ['0', 'j', 'k'], ['0', 'v', 'w']]

pandasを使うなら以下です。

python
1>>> df = pd.DataFrame(a_csv)
2>>> print(df)
3   0  1  2  3  4  5  6  7
40  a  b  c  1  d  e  f  g
51  a  b  c  0  h  i  j  k
62  a  b  c  1  l  m  n  o
73  a  b  c  1  p  q  r  s
84  a  b  c  0  t  u  v  w
9>>> out_df = pd.concat([df[df[3]=='1'].iloc[:,[3,4,5]].set_axis(range(3), axis=1),df[df[3]=='0'].iloc[:,[3,6,7]].set_axis(range(3), axis=1)])
10>>> print(out_df)
11   0  1  2
120  1  d  e
132  1  l  m
143  1  p  q
151  0  j  k
164  0  v  w