Получить направленные элементы в матрице
Предположим, что в моей матрице есть точка интереса NxN. Точка находится в позиции ij. Итак, учитывая индекс ij, существует ли простой способ получить элементы строки, проходящие через ij и к началу координат (которое расположено в середине матрицы)?
Я использую torch и думаю, что использование torch.diag будет первым началом, но на самом деле эта функция не проходит в середину матрицы.
def directionalK(kx,ky, indices):
'''Function that provides the K values at a given direction dictated by the indices'''
kx_grid,ky_grid = torch.meshgrid(kx,kx, indexing='ij')
k_grid = torch.sqrt(kx_grid**2 + ky_grid**2)
k_grid[...,:len(k_grid)//2] *=-1
y,x = indices
diag = x - len(k_grid)//2
Ответы 1
Я думаю, вы могли бы решить эту проблему, используя метод Skimage.draw.line, в котором вы вводите начальную (i,j) и конечную (0,0) координаты, и он вычисляет индексы, принадлежащие строке, которые вы можете использовать для индексации. .
Рабочий пример:
from skimage.draw import line
import numpy as np
import matplotlib.pyplot as plt
arr = np.zeros((100, 100))
i, j = 10, 80
origin = 50, 50
rr, cc = line(*origin, i, j)
arr[rr, cc] = 1
plt.imshow(arr, cmap='gray', interpolation='nearest')
Или, если вы хотите, чтобы линия продолжалась:
arr = np.zeros((100, 100))
rr, cc = line(i, j, 2*origin[0]-i, 2*origin[1]-j)
arr[rr, cc] = 2
plt.imshow(arr, cmap='gray', interpolation='nearest')
Я просто изменил начало координат в центре изображения (50,50) и получил те же результаты, что и в вашем коде.
@Rotacional, извини, я забыл изменить эту строку, когда вставлял код. Хороший улов.
если бы вы сформулировали это в терминах растровой графики, а не «матрицы»…