函数调用后数组不更新值

Question

我需要数组包含我正在生成的(x,y) 坐标的所有可能组合。该数组仍然打印为1s 的数组。

import numpy as np

coordinates = np.ones([1000, 2])

def translate (x,y):
    dx = 5
    dy = 5
    return x + dx, y + dy

for i in range(0, 100):
    for j in range(0, 100):
        (x, y) = translate(i, j)
        coordinates[i, j] = translate(x, y)
        np.append(coordinates, translate(x, y), axis=None)

print(coordinates)

我希望坐标数组在调用 translate 函数后接收正确的值，而不是 1s 的数组。

Answer 1

如果我理解正确，您想要所有可能的(x,y) 坐标，这些坐标可以通过改变x=0..99 和y=0..99 然后分别添加一些dx 和dy 到x 和y .

一种方法，类似于你的，是初始化一个空列表，然后用你定义的翻译函数附加一个元组：

coordinates = []

def translate (x,y):
    dx = 5
    dy = 5
    return x + dx, y + dy

for x in range(0, 100):
    for y in range(0, 100):
        coordinates.append(translate(x,y))

print(coordinates)

但是，这可以改进。意识到您只是将dx 添加到每个x 和dy 到每个y，因此您可以通过删除translate 函数来使用更简单的代码实现相同的结果：

coordinates = []
dx, dy = 5, 5

for x in range(dx, 100+dx):
    for y in range(dy, 100+dy):
        coordinates.append((x,y))

print(coordinates)

这会给你同样的正确答案。