扩展 Python 列表列表中的元素

Question

我需要用不超过一定范围的零填充一个列表列表。我正在计算感染期间减少的利润 (a_x)。感染后，我的所有 3 个系统的利润都必须用零计算。计划期限是 50 年，所以我需要扩展我的列表列表 (tilde_Pi_) 中的元素，以便在每个子列表中分别包含 50 个元素。

# Parameters needed for the calculation
year = np.arange(50)
a_x = list(np.arange(5))
Y_ = [2, 8, 12]
C_ = [620, 2000, 4000]
p = 500
r = 0.03
RC_ = [1000, 5000, 10000]
m_ = [50, 30, 15]
ARC_ = [((RC_[i] * r) / (1 - (1+r) ** (-m_[i]))) for i in range(0, 3)]

# Relevant calculation
tilde_Y_ = [[(Y_[i] * 0.8 ** a_x[a]) for a in a_x] for i in range(0, 3)]
tilde_C_ = [[(C_[i] * 0.8 ** a_x[a]) for a in a_x] for i in range(0, 3)]
tilde_Pi_ = [[((p * tilde_Y_[i][a]) - tilde_C_[i][a] - ARC_[i])
              for a in a_x] for i in range(0, 3)]

我尝试编写 .extend 语法，但出了点问题

tilde_Pi_ = [tilde_Pi_[i].extend([0] * (len(year) - len(tilde_Pi_[i]))) 
             for i in range(0, 3)]

有谁知道如何扩展我的列表列表 (tilde_Pi_) 中的 3 个列表，使其分别包含 50 个元素，前 5 个 (len(a_x)) 计算为以上，余数补零?

Answer 1

这样试试：

for i in range(0, 3):
    tilde_Pi_[i].extend([0] * (len(year) - len(tilde_Pi_[i])))

扩展子列表的方法是listname[index].extend(new_element)

希望对您有所帮助！

扩展 Python 列表列表中的元素

Extend elements in list of lists in Python

python

append

extend