我想使用 numpy python 创建一个由节点名称及其 x y 坐标组成的二维数组

Question

我想为不询问节点的输入创建一个二维数组，并为每个节点从用户接受 2 个值 x 坐标和 y 坐标并生成其二维数组，其中包含 3 列节点、x 坐标、y 坐标。

输入：

enter the number of nodes:3
10 12
12 13
5 6 

输出

[A 10 12
 B 12 13
 C 5 6 ]

Answer 1

定义2个函数：

def foo(node):
    return [node]+[int(i) for i in input('for node: %s: '%node).split()]

def foobar():
    n = input('enter the number of nodes:')
    n = int(n)
    alist=[]
    for i,n in zip(range(n),'ABCDEF'):
        alist.append(foo(n))
    return alist

并根据您的输入获取列表列表：

In [348]: foobar()
enter the number of nodes:3
for node: A: 10 12
for node: B: 12 13
for node: C: 5 6
Out[348]: [['A', 10, 12], ['B', 12, 13], ['C', 5, 6]]

节点名称字符串和数字的混合不能构成一个好的 numpy 数组。

它可以变成一个字符串数组。或 object dtype 数组。或结构化数组。或者更好的是，一个节点名称数组和另一个只有 2 列的 2d 坐标。