python - 来自具有 Python 条件的列表的平均值

Question

list = [[159.2213, 222.2223, 101.2122]
        [359.2222, 22.2210, 301.2144]]

if list[1][0] < list[0][0]:
    avg = (list[1][0] + list[0][0] - 200)/2
else:
    avg = (list[1][0] + list[0][0] + 200)/2

您好!我想对每一列执行此操作并将结果输出到另一个列表中。

Answer 1

修复

您可以循环迭代存在的列数

values = [[159.2213, 222.2223, 101.2122], [359.2222, 22.2210, 301.2144]]
avgs = []
for idx_col in range(len(values[0])):
    if values[1][idx_col] < values[0][idx_col]:
        avg = (values[1][idx_col] + values[0][idx_col] - 200) / 2
    else:
        avg = (values[1][idx_col] + values[0][idx_col] + 200) / 2
    avgs.append(avg)

---

简化

您可以使用zip 一次迭代两行，并简化if/else 条件

avgs = []
for first_row, second_row in zip(*values):
    factor = -1 if second_row < first_row else 1
    avgs.append((first_row + second_row + (200 * factor)) / 2)

---

最好用 numpy

简单语法和最佳性能

import numpy as np

values = np.array(values)
res = values.sum(axis=0) / 2
res += np.where(values[1] < values[0], -100, 100)