Fixes pandas dataframe transpose display error

.pre-commit-config.yaml

@@ -19,6 +19,7 @@ repos:
  hooks:
  - id: ruff
  stages: [commit]
+ args: [--fix, .]
  - id: ruff-format
  stages: [commit]
  - repo: https://github.com/pre-commit/mirrors-mypy

packages/solara-widget-manager8/src

@@ -1 +1 @@
-../solara-widget-manager/src
+../solara-widget-manager/src

packages/solara-widget-manager8/style

@@ -1 +1 @@
-../solara-widget-manager/style
+../solara-widget-manager/style

packages/solara-widget-manager8/tsconfig.json

@@ -1 +1 @@
-../solara-widget-manager/tsconfig.json
+../solara-widget-manager/tsconfig.json

solara/components/datatable.py

@@ -11,7 +11,7 @@
 import solara.lab
 import traitlets
 from solara.lab.hooks.dataframe import use_df_column_names
-from solara.lab.utils.dataframe import df_len, df_records, df_slice
+from solara.lab.utils.dataframe import df_len, df_records, df_slice, df_row_names
 
 from .. import CellAction, ColumnAction
 
@@ -100,12 +100,12 @@ def DataTable(
  i2 = min(total_length, (page + 1) * items_per_page)
 
  columns = use_df_column_names(df)
-
+ rows = df_row_names(df)
  items = []
  dfs = df_slice(df, i1, i2)
  records = df_records(dfs)
  for i in range(i2 - i1):
- item = {"__row__": i + i1} # special key for the row number
+ item = {"__row__": format(dfs, columns, i + 1, rows[i + i1])} # special key for the row number
  for column in columns:
  item[column] = format(dfs, column, i + i1, records[i][column])
  items.append(item)

solara/lab/utils/dataframe.py

@@ -1,4 +1,4 @@
-from typing import List
+from typing import List, Union
 
 
 def get_pandas_major():
@@ -28,6 +28,16 @@ def df_columns(df) -> List[str]:
  raise TypeError(f"{type(df)} not supported")
 
 
+def df_row_names(df) -> List[Union[int, str]]:
+ """Return a list of row names from a dataframe."""
+ if df_type(df) == "vaex" or df_type(df) == "polars":
+ return list(range(df_len(df)))
+ elif df_type(df) == "pandas":
+ return df.index.tolist()
+ else:
+ raise TypeError(f"{type(df)} not supported")
+
+
 def df_slice(df, start: int, stop: int):
  """Return a subset of rows from a dataframe."""
  if df_type(df) == "pandas":