def view_positions(self, indices=None, time=None):
if time is None:
time = 0
res = self.synthetic_store.get(indices=indices, variables=['x', 'y', 'z'])
pylab.figure()
all_x = []
all_y = []
all_z = []
all_c = []
for key in res.keys():
all_x += [res[key]['x'][time]]
all_y += [res[key]['y'][time]]
all_z += [res[key]['z'][time]]
all_c += [self._scalarMap_synthetic.to_rgba(int(key))]
pylab.scatter(self.probe.positions[0, :], self.probe.positions[1, :], c='k')
pylab.scatter(all_x, all_y, c=all_c)
pylab.show()
评论列表
文章目录