python - 使用输入作为数组和字符串

Question

我觉得有一个简单的解决方案，但我有点新。

stat_input= input("Hello Mr. Jenner, what are you interested in tracking today?")

我使用这样的输入，稍后用于调用数据并使用该数据来计算统计数据并生成直方图/正态分布。

效果非常好。以下是一些使用它的示例。

cur.execute('SELECT {} FROM statdata'.format(stat_input))
np.array(stat_input).astype(np.float)

sigma = math.sqrt(np.var(stat_input))

因此，如果我输入thirdmonthdata，它将从我的数据库中提取数据数组并使用它。这很棒。不过，我有一个小问题

我知道 thirdmonthdata 指的是一个数组。由于我正在创建图表，因此我想使用输入作为标题，以便图表标题标识我正在绘制和使用的数据(作为将来的引用)

ax.set_title('stat_input')

这不起作用

ax.set_title(' + stat_input + ') 

这也不是。我希望标题为 Threemonthdata。但是，如果我输入 twomonthdata，我希望它显示 twomonthdata，而不是给我数字数组。

有什么想法吗？

Answer 1

我从来没有玩过psycopg的cursor类(class)。但是，从我所读到的内容来看，这个似乎可以为您完成将字符串转换为名称与引用字符串相同的列表的工作。

那么在字符串被覆盖之前定义另一个变量来存储字符串怎么样？如下

stat_input_title = stat_input.capitalize()
cur.execute('SELECT {} FROM statdata'.format(stat_input))

从此以后，stat_input_title 和 stat_input 可以一起使用，不会发生冲突。

ax.set_title(stat_input_title)