0

答えが簡単に説明できるような問題があります。次のコードを使用して、行と列を含む csv ファイルからインポートしています。

import csv
import os
import glob
import numpy as np

def get_data(filename):
    with open(filename, 'r') as f:
        reader = csv.reader(f)                      
        return list(reader)

all_data = []

path=raw_input('What is the directory?')       
for infile in glob.glob(os.path.join(path, '*.csv')):
     all_data.extend(get_data(infile))
a = np.array(all_data)
col=a[:,[0,1]]
print col

ただし、結果は列ではなく最初の 2 行を生成しますか?

4

1 に答える 1