泛化整个列表列表的公式

Question

我想这样做

firstdates = dates[0]
paystring = []
for i in range(len(payment_months)):
    if payment_months[i] < firstdates[i]:
        paystring.append(0)
    else:
        paystring.append(((payment_months[i] + 12 - firstdates[i]) + 1) % 12)
print(paystring)

但我想对列表列表中的所有日期执行此操作。例如，让我们只关注前两行日期。

print(dates[0:2])

给我

[[8, 8, 7, 7, 6, 5, 4, 4, 11, 10, 10, 8], [7, 6, 3, 2, 2, 2, 1, 0, 11, 10, 10, 9]]

和

print(payment_months)

给我

[8, 7, 6, 5, 4, 3, 2, 1, 12, 11, 10, 9]

然后运行后：

paystring = []
for i in range(len(payment_months)):
    paystring.append([])
    for j in range(len(dates[i])):
        if payment_months[i] < dates[i][j]:
            paystring[i].append(0)
        elif NDD_day[j] > 1:
            paystring[i].append((payment_months[i] + 12 - dates[i][j]) % 12)
        else:
            paystring[i].append( ((payment_months[i] + 12 - dates[i][j]) + 1) % 12)
print(paystring[0:2])

我应该得到这个：

[[1, 0, 0, 0, 0, 0, 0, 0, 2, 2, 1, 2], [2, 2, 4, 4, 3, 2, 2, 2, 2, 2, 1, 1]]

但我得到了这个：

[[0, 0, 1, 1, 2, 3, 4, 5, 0, 0, 0, 0], [0, 1, 4, 5, 5, 5, 6, 8, 0, 0, 0, 0]]

有人可以帮我修复我的代码吗？

这是一个手写的例子，应该很清楚：

我尝试将我的代码更改为：

# Calculate paystring
paystring = []
for i in range(len(payment_months)):
    paystring.append([])
    for j in range(len(dates[i])):
        if payment_months[i] < dates[j][i]:
            paystring[i].append(0)
#         elif NDD_day[j] > 1:
#             paystring[i].append((payment_months[i] + 12 - dates[j][i]) % 12)
        else:
            paystring[i].append( ((payment_months[i] + 12 - dates[j][i]) + 1) % 12)
print(paystring[0:2])

这给了我这个：

[[1, 2, 0, 0, 0, 1, 0, 0, 1, 1, 0, 1], [0, 2, 0, 0, 0, 2, 0, 0, 1, 1, 0, 1]]

希望这会有所帮助，我不知道为什么这会令人困惑，也许我没有最好地解释这一点。

Answer 1

在摆弄了一下之后，最终很明显你为你的payment_months 使用了错误的索引，并且如果你想要dates 中的每一行，你正在迭代错误的range。

设置：

In [35]: dates

Out[35]:
[[8, 8, 7, 7, 6, 5, 4, 4, 11, 10, 10, 8],
 [7, 6, 3, 2, 2, 2, 1, 0, 11, 10, 10, 9]]

In [37]: payment_months
Out[37]: [8, 7, 6, 5, 4, 3, 2, 1, 12, 11, 10, 9]

那么我们有：

paystring = []
for i in range(len(dates)): # len(payment_months) -> len(dates)
    paystring.append([])
    for j in range(len(dates[i])):
        if payment_months[j] < dates[i][j]: # Switched from i to j
            paystring[i].append(0)
        else:
            paystring[i].append( ((payment_months[j] + 12 - dates[i][j]) + 1) % 12) # Switched from i to j

输出：

In [42]: paystring
Out[42]: [[1, 0, 0, 0, 0, 0, 0, 0, 2, 2, 1, 2], [2, 2, 4, 4, 3, 2, 2, 2, 2, 2, 1, 1]]

根据需要。

您实际上可以完全避免这个问题，只需要更 Pythonic 一点（避免索引，只访问元素）。这也更清楚地表明dates 中的每一行与payment_months 是一一对应的：

paystring = []
for date_row in dates:
    paystring.append([])
    for date, payment_month in zip(date_row, payment_months):
        if payment_month < date:
            paystring[-1].append(0)
        else:
            paystring[-1].append( ((payment_month + 12 - date) + 1) % 12)

HTH。