根据给定函数 Lua 的变量选择要写入的数组

Question

我正在尝试运行此代码：

function calcs.grps(Number,ion_color)
    grp .. ion_color .. Y[Number] = (ion_py_mm)
    grp .. ion_color .. Z[Number] = (ion_pz_mm)
end

在 Lua 脚本中，数组已经存在（例如 grp2Y、grp5Z 等），我想使用此函数根据输入的两个变量填充它们。我不断收到错误“'=' expected near '。 ''。我做错了什么？

稍微充实一下：

我在模拟中“飞行”了 120 个离子。这实际上是 12 组 10 个离子。 10 个单独的组由 ion_color 来区分，它是一个从 1 到 12 的整数值。变量“Number”每次只在 1 到 10 之间循环，然后再转到下一个颜色。一旦我填充了这些数组，我想获得每个组的标准偏差。

谢谢！

Answer 1

你不能“构造”变量的名字，但是你可以构造一个索引。使用两级嵌套表。

function calcs.grps(Number,ion_color)
    ion['grp' .. ion_color .. 'Y'][Number] = (ion_py_mm)
    ion['grp' .. ion_color .. 'Z'][Number] = (ion_pz_mm)
end

好吧，实际上你可以，因为所有全局变量都只是 _G 表中的条目，但不要这样做，因为它很糟糕 - 它不可读，会使内容溢出到你不打算的其他函数，等等

Answer 2

您的问题的技术答案是简单地索引 _G，_G 是一个包含所有全局变量的表：

function calcs.grps(Number,ion_color)
    _G['grp' .. ion_color .. Y'][Number] = (ion_py_mm)
    _G['grp' .. ion_color .. 'Z'][Number] = (ion_pz_mm)
end

但我认为更好的问题是，你为什么不这样组织它...

local ions = {
    Red = {
        {
            Y = 0, --Y property
            Z = 0 --Z property
        },
        --Continue your red ions
    },
    NewColor = {
            Y = 0, --Y property
            Z = 0 --Z property
        },
        --Continue this color's ions
    },
    --You get the idea
}

function calcs.grps(color, number)
    ions[color][number].Y = (ion_py_mm)
    ions[color][number].Z = (ion_pz_mm)
end

然后你会传递一种颜色，以及一个数字，指示该颜色的哪个离子

IMO 看起来干净多了。