Как прочитать текстовый файл чисел в массив массивов
В python, используя библиотеку OpenCV, мне нужно создать несколько полилиний. Пример кода для метода полилиний показывает:
cv2.polylines(img,[pts],True,(0,255,255))
У меня все 'pts' выложены в текстовом файле в формате:
x1,y1,x2,y2,x3,y3,x4,y4
x1,y1,x2,y2,x3,y3,x4,y4
x1,y1,x2,y2,x3,y3,x4,y4
Как я могу прочитать этот файл и передать данные в переменную [pts] в вызове метода?
Я пробовал метод np.array(csv.reader(...)) , а также несколько других, примеры которых я нашел. Я могу успешно прочитать файл, но он не в том формате, который нужен методу полилиний. (Я новичок в python, если бы это был C++ или Java, это не было бы проблемой).
Ну да. Но это не работает. Я получаю «TypeError: pts» при запуске.
Ответы 4
Я бы попытался использовать numpy для чтения csv как массива.
from numpy import genfromtxt
p = genfromtxt('myfile.csv', delimiter=',')
cv2.polylines(img,p,True,(0,255,255))
Возможно, вам придется передать аргумент dtype в genfromtext, если вам нужно привести данные к определенному формату.
https://docs.scipy.org/doc/numpy/reference/generated/numpy.genfromtxt.html
Это будет читаться в цифрах, но «p» не в правильном формате. Формат должен быть [[x,y],[x,y],[x,y],[x,y]]
Если вы знаете, что это фиксированное количество элементов в каждой строке:
import csv
with open('myfile.csv') as csvfile:
rows = csv.reader(csvfile)
res = list(zip(*rows))
print(res)
Это фиксированное количество элементов (4 набора координат x, y).
Я знаю, что это не очень красиво, и, вероятно, есть НАМНОГО ЛУЧШИЙ способ сделать это, но он работает. При этом, если бы кто-то мог показать мне лучший способ, это было бы очень признательно.
pointlist = []
f = open(args["slots"])
data = f.read().split()
for row in data:
tmp = []
col = row.split(";")
for points in col:
xy = points.split(",")
tmp += [[int(pt) for pt in xy]]
pointlist += [tmp]
slots = np.asarray(pointlist)
Возможно, вам придется рисовать каждую полилинию отдельно (чтобы расширить ответ @Chris):
from numpy import genfromtxt
lines = genfromtxt('myfile.csv', delimiter=',')
for line in lines:
cv2.polylines(img, line.reshape((-1, 2)), True, (0,255,255))
Если
pts = np.array(csv.read(...)), то вы просто делаетеcv2.polylines(img, pts, ...), верно?