Question

我想绘制的是将矢量从高值变为低值。如果代码从以下开始：

a = [[1, 8, 9, 10],[2, 15, 3, -1],[3,1,6,11],[13,15,5,-2]]
X,Y = np.meshgrid(np.arange(4), np.arange(4))
U = ?
V = ?

从这一点开始，我应该设置向量的U和V个组件。每个点的大小为a[x][y]。我不太了解如何设置U和V以在每个网格点将箭头从高值变为低值。

Answer 1

这是一个解决方案（不需要numpy）：

import itertools as it

a = [[1, 8, 9, 10],[2, 15, 3, -1],[3,1,6,11],[13,15,5,-2]]
rowSize = len(a[0])

maxVal = a[0][0]
maxIndex = 0
minVal = a[0][0]
minIndex = 0

for k, v in enumerate(it.chain(*a)):  # Loop through a flattened list of the values in the array, and locate the indices of the max and min values.
    if v > maxVal:
        maxVal = v
        maxIndex = k    
    if v < minVal:
        minVal = v
        minIndex = k

U = (minIndex % rowSize) - (maxIndex % rowSize)
V = (minIndex / rowSize) - (maxIndex / rowSize)

print U, ",", V

输出

2 , 2

请注意，当您的示例中存在两个相等的最大值时，您尚未定义所需的行为。上面的代码使用“first”（最左上角）作为真正的最大值，并忽略所有其他值。

说明：

我把列表弄平了（这意味着我会像书上的文字那样读取值 - 首先是第一行，然后是第二行，然后是第三行）。每个值都有一个索引，如下所示：

0  1  2  3
4  5  6  7
8  9  10 11
12 13 14 15

例如，第二行和第三列中的值将获得6的索引，因为如果您像书本一样读取数组，则它是第7个值。

最后，当我们找到最大值或最小值的索引时，我们需要从1D索引中获取2D坐标。因此，我们可以使用mod运算符（％）来获取x值。

例如，6 % 4 = 2，X = 2（第3列）

要获得Y值，我们使用整数除法运算符（/）。

例如，6 / 4 = 1，Y = 1（第二行）

U和V的公式只是取最大值和最小值的X和Y值，然后减去它们以得到矢量坐标，如下所示：

U = xMin - xMax
V = yMin - yMax

如果你想知道，“为什么他没有使用我开始使用的meshgrid解决方案”，有两个原因：一，使用像numpy这样的非标准库通常是不可取的，如果有一个简单的话没有非标准库解决问题的方法，以及两个，如果你需要处理大型数组，生成大型网格网格可能会花费时间/内存。

选择最短矢量的解决方案：

import itertools as it

a = [[1, 8, 9, 10],[2, 15, 3, -1],[3,1,6,11],[13,15,5,-2]]
rowSize = len(a[0])

values = sorted(enumerate(it.chain(*a)), key=lambda x:x[1])  # Pair each value with its 1D index, then sort the list.

minVal = values[0][1]
maxVal = values[-1][1]

maxIndices = map(lambda x:x[0], filter(lambda x:x[1]==maxVal, values))  # Get a list of all the indices that match the maximum value
minIndices = map(lambda x:x[0], filter(lambda x:x[1]==minVal, values))  # Get a list of all the indices that match the minimum value

def getVector(index1, index2, rowSize):  # A function that translates a pair of 1D index values to a "quiver vector"
    return ((index1 % rowSize) - (index2 % rowSize), (index1 / rowSize) - (index2 / rowSize))

vectors = [getVector(k2, k1, rowSize) for k1, k2 in it.product(maxIndices, minIndices)]  # produce a list of the vectors formed by all possible combinations of the 1D indices for maximum and minimum values

U, V = sorted(vectors, key=lambda x:(x[0]*x[0] + x[1]*x[1])**0.5)[0]

print U, ",", V

输出

2 , 0

箭矢矢量从高值到低值

1 个答案:

输出

说明：

选择最短矢量的解决方案：

输出