Pythonの関数において*argの引数を動的に設定する方法

Question

前提・実現したいこと

numpy.meshgridにて格子点を発生させる際に、変数の数が動的に変更される（例えばユーザーの入力次第で2次元だったり3次元だったりする）場合に、変数の数に応じてmeshgridを使えるようにしたい。
（例）ユーザーが変数の数（および水準）を自由に設定して、設定された変数に応じて自動でグリッド点を発生させたい。

できること

import numpy as np
a = np.arange(0, 10, 1)
b = np.arange(2, 12, 1)
c = np.arange(5, 10, 1)
np.meshgrid(a,b,c)

当然上のようにすると正しくgrid点を発生できますが、例えば変数の数をユーザーが以下のように入力し
（実際は水準含めて自分で入力できるようにしますが）

変数の選択

variables_num = int(input('the number of variable'))
variable_selected = variable_list[:variables_num]

その選択された変数の数に応じてmeshgridを使う方法がわかりません。

試したこと

np.meshgrid(variable_selected)

当たり前ですが望む格子点は得られません。

以上を通して素朴な疑問として*argに引数を動的に入力できる方法はないか気になりました。
何か良い方法があれば教えていただけますと幸いです。

（併せて本来の目的である何かユーザーの入力に応じて自動で格子点を生成できるより良い方法があれば教えていただけますと幸いです。*argに引数を動的に入力できる方法以外でも構いません。）

よろしくお願いいたします。

補足情報

Python 3.7を使用

Answer 1

np.meshgrid(*variable_selected)

Comments

[UZK]

ありがとうございます！！そんな単純なことだったのですね…検索が下手くそな初心者でしたがご回答いただけて大変助かります！！

Answer 2

*を使って引数を展開してあげれば良いかと思います。

import numpy as np
arr = np.array(range(21)).reshape([7,3])
print(arr)
# [[ 0  1  2]
#  [ 3  4  5]
#  [ 6  7  8]
#  [ 9 10 11]
#  [12 13 14]
#  [15 16 17]
#  [18 19 20]]

def v_mesh(variable_list):
    variables_num = int(input('the number of variable'))
    variable_selected = variable_list[:variables_num]
    return np.meshgrid(*variable_selected)
print(v_mesh(arr))
# input -> 2の場合
# [array([[0, 1, 2],
#        [0, 1, 2],
#        [0, 1, 2]]), array([[3, 3, 3],
#        [4, 4, 4],
#        [5, 5, 5]])]

Comments

[UZK]

ありがとうございます。何故そこに行きつけなかったか自分でも不思議です…とても助かりました！！