1

几个小时以来一直在敲我的头，这是我能得到的最干净的地方，但它似乎可以变得更干净。

center1_x = np.mean([q[0] for q in coordinates if q[-1]==0])
center2_x = np.mean([q[0] for q in coordinates if q[-1]==1])
center3_x = np.mean([q[0] for q in coordinates if q[-1]==2])
centers_x=[center1_x,center2_x,center3_x]

坐标是子列表的列表，其中每个子列表包含 [0] 处的 X 坐标、[1] 处的 y 坐标和 [-1] 处的 bin 编号，即：

[[338.766845703125, 285.3931884765625, 0], [198.39230346679688, 184.1259307861328, 1], [356.7226867675781, 281.6376953125, 0], [200.93394470214844, 194.82655334472656, 1], [228.03591918945312, 297.79864501953125, 2], [226.06724548339844, 295.5475769042969, 2], [246.51475524902344, 280.373779296875, 2], [201.25830078125, 184.09994506835938, 1]]
4

1 回答 1

3

尝试

centers_x = [np.mean([q[0] for q in coordinates if q[-1]==n]) for n in range(3)]
于 2012-08-14T00:52:23.270 回答