Update packages/jspsych/src/modules/data/DataCollection.ts

Co-authored-by: bjoluc <mail@bjoluc.de>
This commit is contained in:
Josh de Leeuw 2022-03-11 12:09:26 -05:00 committed by GitHub
parent cafc6a1f65
commit 9180ab9671
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -135,14 +135,11 @@ export class DataCollection {
}
filterColumns(columns: Array<string>) {
const filtered_data = [];
for (const trial of this.trials) {
const filtered_columns = Object.fromEntries(
columns.filter((key) => key in trial).map((key) => [key, trial[key]])
);
filtered_data.push(filtered_columns);
}
return new DataCollection(filtered_data);
return new DataCollection(
this.trials.map((trial) =>
Object.fromEntries(columns.filter((key) => key in trial).map((key) => [key, trial[key]]))
)
);
}
select(column) {