本文整理汇总了Python中xray.Dataset.squeeze方法的典型用法代码示例。如果您正苦于以下问题:Python Dataset.squeeze方法的具体用法?Python Dataset.squeeze怎么用?Python Dataset.squeeze使用的例子?那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类xray.Dataset
的用法示例。
在下文中一共展示了Dataset.squeeze方法的2个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: test_squeeze
# 需要导入模块: from xray import Dataset [as 别名]
# 或者: from xray.Dataset import squeeze [as 别名]
def test_squeeze(self):
data = Dataset({'foo': (['x', 'y', 'z'], [[[1], [2]]])})
for args in [[], [['x']], [['x', 'z']]]:
def get_args(v):
return [set(args[0]) & set(v.dimensions)] if args else []
expected = Dataset({k: v.squeeze(*get_args(v))
for k, v in data.variables.iteritems()})
self.assertDatasetIdentical(expected, data.squeeze(*args))
# invalid squeeze
with self.assertRaisesRegexp(ValueError, 'cannot select a dimension'):
data.squeeze('y')
示例2: test_squeeze
# 需要导入模块: from xray import Dataset [as 别名]
# 或者: from xray.Dataset import squeeze [as 别名]
def test_squeeze(self):
data = Dataset({'foo': (['x', 'y', 'z'], [[[1], [2]]])})
# squeeze everything
expected = Dataset({'y': data['y'], 'foo': data['foo'].squeeze()})
self.assertDatasetEqual(expected, data.squeeze())
# squeeze only x
expected = Dataset({'y': data['y'], 'foo': (['y', 'z'], data['foo'].data[0])})
self.assertDatasetEqual(expected, data.squeeze('x'))
self.assertDatasetEqual(expected, data.squeeze(['x']))
# squeeze only z
expected = Dataset({'y': data['y'], 'foo': (['x', 'y'], data['foo'].data[:, :, 0])})
self.assertDatasetEqual(expected, data.squeeze('z'))
self.assertDatasetEqual(expected, data.squeeze(['z']))
# invalid squeeze
with self.assertRaisesRegexp(ValueError, 'cannot select a dimension'):
data.squeeze('y')