Как мне прочитать текстовый файл чисел в массив массивов - PullRequest
0 голосов
/ 09 апреля 2019

В 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 (...)), а также несколько других примеров, которые я нашел. Я могу успешно прочитать файл, но он не в том формате, который хочет метод polylines. (Я новичок, когда дело доходит до Python, если бы это был C ++ или Java, это не было бы проблемой).

Ответы [ 4 ]

0 голосов
/ 11 апреля 2019

Возможно, вам понадобится нарисовать каждую полилинию отдельно (чтобы расширить ответ @ 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))
0 голосов
/ 09 апреля 2019

Если вы знаете, что это фиксированное количество элементов в каждой строке:

import csv

with open('myfile.csv') as csvfile:
    rows = csv.reader(csvfile)
    res = list(zip(*rows))
    print(res)
0 голосов
/ 10 апреля 2019

Я знаю, что это не красиво, и, вероятно, есть НАМНОГО ЛУЧШЕГО способа сделать это, но это работает. При этом, если бы кто-то мог показать мне лучший путь, это было бы очень ценно.

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)
0 голосов
/ 09 апреля 2019

Я бы попробовал использовать 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

...