簡體   English   中英

從 csv 文件中讀取兩個 Numpy Arrays

[英]Reading Two Numpy Arrays from csv file

所以我想從 csv 文件中讀取兩個二維數組 A 和 B 作為 numpy-Array
我有以下帶有“;”的 csv 文件分離兩個 arrays:

A;B  
1,1,2,2;3,3,4,4  
6,7,3,7;3,5,3,5  
1,8,5,3;6,1,7,5  

結果應該是這樣的

A = [[1, 1, 2, 2], [6, 7, 3, 7], [1, 8, 5, 3]]   
B = [[3, 3, 4, 4], [3, 7, 3, 7], [6, 1, 7, 5]]  

現在我應該怎么做,我嘗試了很多 loadtxt 和 genfromtxt 但做不到

你試過csv嗎? 我沒有發布整個代碼,而是這樣的:

import csv

with open('fileName.csv') as file:
    csv_reader =  csv.reader(file, delimiter=',')
    for row in csv_reader:
        #do sth

這應該可行,可能有更多的 numpy 方式,但這就是我制作數組的方式:

import numpy as np
import pandas as pd
A=[]
B=[]
df1=pd.read_csv('numpy.csv', sep=";") 
for x in range(len(df1.A)):
   A.append(df1.A[x].split(','))
for x in range(len(df1.B)):
   B.append(df1.B[x].split(','))
A=np.array(A).astype(np.int)
B=np.array(B).astype(np.int)

A
#array([[1, 1, 2, 2],
#       [6, 7, 3, 7],
#       [1, 8, 5, 3]])

B
Out[251]: 
#array([[3, 3, 4, 4],
#       [3, 5, 3, 5],
#       [6, 1, 7, 5]])

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM