From 38c7e6a8e40bca3fafc3be8f16609c617b9f59f0 Mon Sep 17 00:00:00 2001 From: Martin Diehl Date: Tue, 28 Nov 2023 20:52:49 +0100 Subject: [PATCH] consistent code --- python/damask/_vtk.py | 22 ++++++++++------------ 1 file changed, 10 insertions(+), 12 deletions(-) diff --git a/python/damask/_vtk.py b/python/damask/_vtk.py index 08d8a1819..aa1e6fcbf 100644 --- a/python/damask/_vtk.py +++ b/python/damask/_vtk.py @@ -558,20 +558,18 @@ class VTK: """ cell_data = self.vtk_data.GetCellData() - for a in range(cell_data.GetNumberOfArrays()): - if cell_data.GetArrayName(a) == label: - try: - return vtk_to_numpy(cell_data.GetArray(a)) - except AttributeError: - vtk_array = cell_data.GetAbstractArray(a) # string array + if label in [cell_data.GetArrayName(a) for a in range(cell_data.GetNumberOfArrays())]: + try: + return vtk_to_numpy(cell_data.GetArray(label)) + except AttributeError: + vtk_array = cell_data.GetAbstractArray(label) # string array point_data = self.vtk_data.GetPointData() - for a in range(point_data.GetNumberOfArrays()): - if point_data.GetArrayName(a) == label: - try: - return vtk_to_numpy(point_data.GetArray(a)) - except AttributeError: - vtk_array = point_data.GetAbstractArray(a) # string array + if label in [point_data.GetArrayName(a) for a in range(point_data.GetNumberOfArrays())]: + try: + return vtk_to_numpy(point_data.GetArray(label)) + except AttributeError: + vtk_array = point_data.GetAbstractArray(label) # string array try: # string array