将循环替换为列表理解，而不是循环获取函数以在列表理解中返回新数组

Question

基本上，在我有如下所示的代码之前，我试图避免遍历大数组：

for rows in book:
        bs = []
        as = []
        trdsa = []
        trdsb = []
        for ish in book:
            var = (float(str(ish[0]).replace(':',"")) - float(str(book[0]).replace(':',"")))
            if var < .1 and var > 0 :
                bs.append(int(ish[4]))
                as.append(int(ish[5]))
                trdsa.append(int(ish[-2]))
                trdsb.append(int(ish[-1]))
                time = ish[0]
            bflow = sum(numpy.diff(bs))
            aflow = sum(numpy.diff(as))
            OFI = bflow - aflow - sum(trdsb) + sum(trdsa)
            OFIlist.append([time,bidflow,askflow,OFI])

我不想在列表中循环两次，因为它会消耗太多时间。我想我可以做一个列表理解，但我不确定我是否在正确的轨道上

OFIcreate(x,y):
    bs = []
    as = []
    trdsa = []
    trdsb = []
    var = (float(str(y[0]).replace(':',"")) - float(str(x[0]).replace(':',"")))
    if var < .1 and var >= 0 :
        bs.append(int(ish[4]))
        as.append(int(ish[5]))
        trdsa.append(int(ish[-2]))
        trdsb.append(int(ish[-1]))
        time = ish[0]
    bflow = sum(numpy.diff(bs))
    aflow = sum(numpy.diff(as))
    OFI = bflow - aflow - sum(trdsb) + sum(trdsa)
    OFIlist.append([time,bidflow,askflow,OFI])
    return OFIlist
    OFIc = [ OFIcreate(x,y) for x in book for y in book)

问题是我想遍历列表并将 var>=0 和 var <.1 的所有实例分组，然后将值附加到新列表中。我现在拥有它的方式我不认为它会这样做，因为它只会继续创建长度为一的列表。关于我如何实现这一目标的任何想法？或者更确切地说，如何使第一个代码块更有效率？

Answer 1

虽然列表推导式的解释确实比常规循环更快，但它们不能适用于所有内容。我不认为你可以用列表理解来取代你的主for循环。但是，可能还有一些改进的余地：

• 您可以通过列表理解来构建time列表。

  time = [ish[0] for ish in book]
  

• 您可以通过列表推导计算var列表并将其转换为np.array。

  var = np.array([t.replace(':',',') for t in time], dtype=float)
  var -= float(str(book[0]).replace(":", ","))
  

• 你可以为 bs 、as 构建 4 个 numpy int 数组（你需要重命名，as是一个 Python 关键字）......

• 然后，您可以过滤您的bs...具有花哨索引的数组：

  bs_reduced = bs[(var < 0.1) & (var >=0)]
  

Answer 2

我不想在列表中循环两次，因为它会消耗太多时间。我想我可以做一个列表理解，但我不确定我是否在正确的轨道上

应该不会。列表推导除了循环遍历给定的列表外什么都不做，因此它应该没有明显的区别。