Review comment

Signed-off-by: martinRenou <[email protected]>

Author: martinRenou

ipydatagrid/datagrid.py

@@ -896,13 +896,10 @@ def _get_row_index_of_primary_key(self, value):
                 "as the primary key."
             )
 
-        # TODO Is there a better way for this?
-        row_indices = [
-            idx
-            for idx, row in self._data["data"].iterrows()
-            if all(row[key[j]] == value[j] for j in range(len(key)))
-        ]
-        return row_indices
+        df = self._data["data"]
+        return pd.RangeIndex(len(df))[
+            (df[key] == value).all(axis="columns")
+        ].to_list()
 
     @staticmethod
     def _get_cell_value_by_numerical_index(data, column_index, row_index):