我有一個這樣的 csv 文件： l1 p1 p2 p3 ... p7841 ↑ ← - - r o w - - →. c ← - - r o w - - →. o ← - - r o w - - →. l ← - - r o w - - →27454 ↓ ← - - r o w - - →# rows: 27454 cols: 784我想使用csv 模塊將它轉換成兩個像這樣的 numpy 數組......l=[ ←, c, o, l, → ] #size: 27454并且r = [ [ ←, -, -, r, o, w, -, -, → ], [ ←, -, -, r, o, w, -, -, → ], [ ←, -, -, r, o, w, -, -, → ] ]#size : (27454, 784)此外，我想將r 數組#shape: (27454, 784)拆分為#shape: (27454, 28,28)如何獲得這些數組？
1 回答
Cats萌萌
TA貢獻1805條經驗 獲得超9個贊
做這個：
（路徑是您的 csv 文件的路徑）
fields = []
rows = []
with open(path, 'r') as csv_f:
csv_reader = csv.reader(csv_f)
rows = [r for r in csv_reader]
l = np.array([r[0] for r in rows])
i_ = np.array([r[1:] for r in rows])
i = np.array([np.array_split(r, 28) for r in i_])
i = i.astype(int)
print(labels.shape)
print(images.shape)
print(l.shape)
輸出：
(27454,)
(27454, 784)
(27454, 28, 28)
添加回答
舉報
0/150
提交
取消