請問大家有沒有簡單的方法可以將data檔讀入numpy的2D array中:
data檔是這樣:
1 : 100 200 300
2 : 101 201 301
...
檔案中有不要的 ":" .需要拿掉,我試了以下方法是可以的,不知道還有沒有其他簡潔的方法:
import numpy as np
import pandas as pd
df = pd.read_csv(filename, header=None, sep="\s+")
data=df.to_numpy(copy=True)
data2=np.delete(data,1,1)
這樣好像就可以得到純數字的2D array.
data2=[[1 100 200 300]
[2 101 200 300]
...
[ ]]
請問有其他簡單一點的寫法嗎?
已邀請的邦友 {{ invite_list.length }}/5
test.csv
1 : 100 200 300
2 : 101 201 301
3 : 100 200 300
4 : 101 201 301
5 : 100 200 300
6 : 101 201 301
test.py
import numpy
data = numpy.genfromtxt('test.csv',usecols=(0,2,3,4))
print(data)
小弟獻醜一下：
import pandas as pd
location = ('test.csv')
df = pd.read_csv(location, header = None)
df = df[0].str.replace(':', "")
df
iThome鐵人賽
看影片追技術