动态设置变量名的合适时机？

Question

编辑：原来答案是明确的“不”。但是，我仍在努力用正确数量的条目填充列表。

---

我一直在搜索 StackOverflow，我一直发现动态设置变量名不是一个好的解决方案。但是，我想不出另一种方法来解决这个问题。

我有一个从pandas 创建的DataFrame（从 excel 中读取），其中包含带有字符串标题和整数条目的列，以及一列具有从 1 到 52 顺序增加的数字（我们称之为周）。我想要做的是创建单独的列表，每个列表都以列标题命名，条目是出现列出整数次数的周数。

这对于几列来说很简单，只需手动创建列表名称，但随着列数的增加，这可能会有点失控。

残酷的解释，这是我能想到的最好的解释。希望一个简化的例子可以澄清。

week  str1    str2    str3   
1       8       2        5        
2       1       0        3    
3       2       1        1 

期望的输出：

str1_count = [1, 1, 1, 1, 1, 1, 1, 1, 2, 3, 3] # eight 1's, one 2, and two 3's
str2_count = [1, 1, 3]                         # two 1's, one 3
str3_count = [1, 1, 1, 1, 1, 2, 2, 2, 3]       # five 1's, three 2's, one 3

到目前为止我所拥有的：

results = {}
df = pd.DataFrame(from_csv(...., sep = ","))
for key in df:
    for i in df[key]
        results[key] = i # this only creates a list with the int value of the most recent i

Answer 1

那么，像这样吗？

import collections
import csv
import io

reader = csv.DictReader(io.StringIO('''
week,str1,str2,str3
1,8,2,5
2,1,0,3
3,2,1,1
'''.strip()))

data = collections.defaultdict(list)
for row in reader:
    for key in ('str1', 'str2', 'str3'):
        data[key].extend([row['week']]*int(row[key]))

from pprint import pprint
pprint(dict(data))

# Output:
{'str1': ['1', '1', '1', '1', '1', '1', '1', '1', '2', '3', '3'],
 'str2': ['1', '1', '3'],
 'str3': ['1', '1', '1', '1', '1', '2', '2', '2', '3']}

注意：Pandas 非常适合处理数据并对其进行一些有趣的操作，但如果您只需要一些简单的东西，则不需要它。这是其中一种情况。