csvファイルから特定の行だけ取り出して、ほかのcsvファイルの特定の行と連結させ新しいcsvファイルを作りたい

csvファイルの特定の行を連結させて新しいファイルを作りたいです

例としては
1,2,3,4
5,5,3,2
1,2,4,6
3,4,5,1
2,1,3,4

と

2,4,6,7
3,4,5,6
3,2,3,5
1,2,2,1
3,5,6,7

のようなcsvファイルの1行目を抜き出し
結果として

1,2,3,4
2,4,6,7

のようなcsvファイルを作りたいです

下記のプログラムを動かしたところ

1
2
3
4
2
4
6
7

のようなファイルができてしまいました

python
1def mk_file(EMG_data_file):
2    count = 0
3    EMG_file = glob.glob(EMG_data_file)
4    for f_name in EMG_file:
5        df = pd.read_csv(f_name, header=None)
6        data = np.array(df)
7
8        if(count == 0):
9            EMG1 = data[0]
10            EMG2 = data[1]
11            EMG3 = data[2]
12            EMG4 = data[3]
13            EMG5 = data[4]
14
15        elif(count != 0 and count <= len(EMG_file)-1):
16            EMG1 = np.append(EMG1, data[0])
17            EMG2 = np.append(EMG2, data[1])
18            EMG3 = np.append(EMG3, data[2])
19            EMG4 = np.append(EMG4, data[3])
20            EMG5 = np.append(EMG5, data[4])
21
22            if(count == len(EMG_file)-1):
23                write_EMGdata(r"c:\project\EMG_data\EMG1.csv", EMG1)
24                write_EMGdata(r"c:\project\EMG_data\EMG2.csv", EMG2)
25                write_EMGdata(r"c:\project\EMG_data\EMG3.csv", EMG3)
26                write_EMGdata(r"c:\project\EMG_data\EMG4.csv", EMG4)
27                write_EMGdata(r"c:\project\EMG_data\EMG5.csv", EMG5)
28
29                return EMG1, EMG2, EMG3, EMG4, EMG5
30
31        count += 1

Zuishin

2019/10/14 09:04

「ある」とか「特定の」とかが多すぎてよくわかりませんが、結局一つのテキストファイルの一行目ともう一つのテキストファイルの二行目を出力したいということではありませんか？

hanayamada

2019/10/14 09:11

実際には240行5列のcsvファイルの同じ行だけを新しく同じcsvファイルとして出力したいということです修正しておきます

Zuishin

2019/10/14 09:17 編集

結果が二行になっていますが、二つのテキストファイルの一行目だけということではなく、最後まで交互に出すということですか？その場合、ファイルの行数が違っていたらどうなりますか？これは単なるテキストファイルの処理で、CSV かどうかも関係ないように思います。

hanayamada

2019/10/14 13:37

そういうことですファイルの行数が違う場合はありません csvにて保存しているファイルについての質問だったためこのような質問の仕方をしました分かりずらく申し訳ありません

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

回答1件

ベストアンサー

Pandas も NumPy も使わない版ですが、
以下の前提で、書いてみました。

全部のファイルに同じ行数という前提
CSV のデータは "" と改行がない前提

python
1#!/usr/bin/env python
2
3import glob
4
5def mk_file(emg_files_glob):
6  emg_files = glob.glob(emg_files_glob)
7
8  data = []
9  for emg_file in emg_files:
10    with open(emg_file) as f:
11      for i, line in enumerate(f.readlines()):
12        if i >= len(data):
13          data.append([])
14
15        data[i].append(line)
16
17  for i, lines in enumerate(data):
18    out_file = 'out/EMG%d.csv' % (i + 1)
19    with open(out_file, 'w') as f:
20      f.writelines(lines)
21
22
23
24mk_file('in/*.csv')
25

出したい結果はこうですよね？

txt
1$ head out/*.csv
2==> out/EMG1.csv <==
31,2,3,4
42,4,6,7
5
6==> out/EMG2.csv <==
75,5,3,2
83,4,5,6
9
10==> out/EMG3.csv <==
111,2,4,6
123,2,3,5
13
14==> out/EMG4.csv <==
153,4,5,1
161,2,2,1
17
18==> out/EMG5.csv <==
192,1,3,4
203,5,6,7
21
22$ head in/*.csv
23==> in/a.csv <==
241,2,3,4
255,5,3,2
261,2,4,6
273,4,5,1
282,1,3,4
29
30==> in/b.csv <==
312,4,6,7
323,4,5,6
333,2,3,5
341,2,2,1
353,5,6,7
36