将输入同时用作数组和字符串

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))

因此，如果我键入threemonthdata，它将从数据库中提取数据数组并使用它。太棒了。然而，我有一个小问题

我知道threemonthdata是指数组。因为我正在创建图表，所以我希望使用输入作为标题，以便图表标题标识我正在绘制和使用的数据（作为将来的参考）

ax.set_title('stat_input')

这不管用

ax.set_title(' + stat_input + ') 

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

有什么想法吗

Answer 1

看起来您面临的问题是，您正在传递set_title()字符串'stat_input'，而不是变量stat_input。您可能只需要使用：

ax.set_title(stat_input) 

Answer 2

我从来没有玩过psycopg的^{}类。但是，据我所知，这一个似乎为您完成了将您的字符串转换成一个名称与引用字符串相同的列表的工作

因此，在重写字符串之前，如何定义另一个变量来存储它呢？如下

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)