Skip to content

Commit ccbe5ad

Browse files
committed
Dataframe: Use snapshots API for iterating
To support Streaming workflows
1 parent 7b5662a commit ccbe5ad

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

src/binding/python/openpmd_api/DataFrame.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -113,11 +113,11 @@ def iterations_to_dataframe(series, species_name):
113113

114114
df = pd.concat(
115115
(
116-
series.iterations[i]
116+
iteration
117117
.particles[species_name]
118118
.to_df()
119119
.assign(iteration=i)
120-
for i in series.iterations
120+
for i, iteration in series.snapshots().items()
121121
),
122122
axis=0,
123123
ignore_index=True,
@@ -170,12 +170,12 @@ def iterations_to_cudf(series, species_name):
170170
cdf = cudf.concat(
171171
(
172172
cudf.from_pandas(
173-
series.iterations[i]
173+
iteration
174174
.particles[species_name]
175175
.to_df()
176176
.assign(iteration=i)
177177
)
178-
for i in series.iterations
178+
for i, iteration in series.snapshots().items()
179179
),
180180
axis=0,
181181
ignore_index=True,

0 commit comments

Comments
 (0)