异质列表切片

Question

假设我想做一个价格与不同产品数量的基本情节。我的输入数据如下所示：

P = np.array((1.75, 250), (1.59, 355), (1.25, 330))

现在我可以用P[:, 0]获得价格。但现在我想添加品牌名称，如下所示：

P = ((1.75, 250, "Store 1"), (1.59, 355, "Store 2"), (1.25, 330, "Store 3"))

因为我在这里混合了字符串和数字，所以我不能再使用np.array，因为我不想将数字作为字符串。但是现在我不能再P[:, 0]来获得价格了（它给了我一个TypeError: tuple indices must be integers, not tuple）。

那么还有一种简单的方法可以在这样的异构列表中访问价格吗？我知道我可以把它们放在不同的列表中，但是我这样可以学到一些关于Python的新东西。

Answer 1

您可以使用列表理解：

>>> P = ((1.75, 250, "Store 1"), (1.59, 355, "Store 2"), (1.25, 330, "Store 3"))
>>> [product[0] for product in P]
[1.75, 1.59, 1.25]

Answer 2

使用struct数组; start here

Answer 3

使用map和lambda表达式怎么样？

>>> P = ((1.75, 250, "Store 1"), (1.59, 355, "Store 2"), (1.25, 330, "Store 3"))
>>> price = map(lambda x: x[0], P)
>>> price
[1.75, 1.59, 1.25]