代码之家 › 专栏 › 技术社区 › Rotacional

获取矩阵中的方向元素

torch signal-processing matrix python

Rotacional · 技术社区 · 2 年前

假设我的矩阵中有一个感兴趣的点,即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 回复 | 直到 2 年前

Warren Weckesser 2 年前

我想你可以用 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')

推荐文章

AstralHex · 矩阵乘法代码工作不正常

1 年前

Baba Dan Constantin · SSE4.1在矩阵4x4乘法上比SSE3慢?

1 年前

Max · 用两列中的特定值对识别R中的数据帧行

1 年前

RobertF · 如何将函数应用于矩阵的每个元素?

1 年前

Leprechault · 使用数据帧创建对象类“matrix”“array”

1 年前

Landers · 将矩阵转换为包含所有值的列表

2 年前

P_B · 如何从矩阵中减去均值向量

2 年前

Rotacional · 获取矩阵中的方向元素

2 年前

ImRobb · 为什么int**m不等于int m[][]?[关闭]

2 年前

Suyash Nachankar · 为什么矩阵的所有行都与最后一行相同?

2 年前