打印带标题的表格时替代 locals()

Question

[Python 3.1]

编辑：原始代码中的错误。

我需要打印一张表格。第一行应该是标题，由制表符分隔的列名称组成。以下行应包含数据（也以制表符分隔）。

为了澄清起见，假设我有“速度”、“功率”、“重量”列。我最初在我之前提出的相关问题的帮助下编写了以下代码：

column_names = ['speed', 'power', 'weight']

def f(row_number):
  # some calculations here to populate variables speed, power, weight
  # e.g., power = retrieve_avg_power(row_number) * 2.5
  # e.g., speed = math.sqrt(power) / 2
  # etc.
  locals_ = locals()
  return {x : locals_[x] for x in column_names}

def print_table(rows):
  print(*column_names, sep = '\t')
  for row_number in range(rows):
    row = f(row_number)
    print(*[row[x] for x in component_names], sep = '\t')

但后来我了解到我应该避免使用locals() 如果可能的话。

现在我被困住了。我不想多次键入所有列名称的列表。我不想依赖这样一个事实：我在 f() 中创建的每个字典都可能以相同的顺序迭代其键。而且我不想使用 locals()。

请注意，函数 print_table() 和 f() 还执行许多其他操作；所以我必须把它们分开。

代码应该怎么写呢？

Answer 1

class Columns:
    pass

def f(row_number):
    c = Columns()
    c.power = retrieve_avg_power(row_number) * 2.5
    c.speed = math.sqrt(power) / 2
    return c.__dict__

这还允许您指定哪些变量作为列，而不是在函数中临时存在。

Answer 2

您可以使用 OrderedDict 来修复字典的顺序。但据我看来，这甚至没有必要。您始终从 column_names 列表中获取键（除了最后一行，我认为这是一个拼写错误），因此值的顺序将始终相同。

Answer 3

locals() 的替代方法是使用 inspect 模块

import inspect

def f(row_number):
    # some calculations here to populate variables speed, power, weight
    # e.g., power = retrieve_avg_power(row_number) * 2.5
    # e.g., speed = math.sqrt(power) / 2
    # etc.
    locals_ = inspect.currentframe().f_locals
    return {x : locals_[x] for x in column_names }