Python：根据输入从列表中获取项目

Question

我很欣赏这可能不是直接可能的，所以我会对您在一般情况下如何解决此问题感兴趣。 我有一个看起来像这样的列表项， [(array,time),(array,time)...]数组是一个numpy数组，可以具有任何n×m的维。 看起来像array[[derivatives dimension1],[derivatives dimension 2] ...]

从列表中，我希望一个函数创建两个列表，其中将包含传递给它的位置处的所有值。 然后可以将这些用于绘图。 我可以考虑使用替代数据结构来实现此目的的方法，但是不幸的是，这是没有选择的。

本质上我想要的是

def f(list, pos1, pos2):
    xs = []
    ys = []
    for i in list:
        ys.append(i pos1)
        xs.append(i pos2)
    return xs, ys

i pos1等于i[n][m]真正的问题是当它为1乘1时，我不能只传递整数。 任何建议都很好，很抱歉，我想弄清楚这篇文章有点长。 谢谢

Answer 1

如果我正确地理解了您的问题，则本质上您想从列表列表中选择索引，然后从该选择中创建新列表。

从列表列表中选择索引非常简单，特别是如果您有固定数量的选择：

parts = [(item[pos1], item[pos2]) for item in list]

使用内置的zip（）函数，从这些选择中创建新列表也相当容易：

separated = zip(*parts)

您可以通过使用生成器表达式而不是最终函数中的列表理解来进一步减少内存使用：

def f( list, pos1, pos2 ):
    partsgen = ((item[pos1], item[pos2]) for item in list)
    return zip(*partsgen)

运作方式如下：

>>> f( [['ignore', 'a', 1], ['ignore', 'b', 2],['ignore', 'c', 3]], 1, 2 )
[('a', 'b', 'c'), (1, 2, 3)]

更新：重新阅读问题和评论后，我意识到这有点过分简化了。 但是，当您交换pos1和pos2以便对包含的数组进行适当的索引编制时，总体思路仍然应该起作用。

Answer 2

如果我了解您的问题，则类似以下内容的操作应该轻松快捷，尤其是如果您需要多次执行以下操作：

z = np.dstack([ arr for arr, time in lst ])
x, y = z[pos1], z[pos2]

例如：

In [42]: a = arange(9).reshape(3,3)
In [43]: z = np.dstack([a, a*2, a*3])
In [44]: z[0,0]
Out[44]: array([0, 0, 0])
In [45]: z[1,1]
Out[45]: array([ 4,  8, 12])
In [46]: z[0,1]
Out[46]: array([1, 2, 3])