是否可以在material-ui DataGrid中初始化复选框选择？

import React from 'react';
import { DataGrid, GridColDef } from '@material-ui/data-grid';

const columns: GridColDef[] = [
  { field: 'id', headerName: 'ID', width: 70 },
  { field: 'name', headerName: 'Name', width: 130 },
  { field: 'age', headerName: 'Age', type: 'number', width: 90 },
];

const rows = [
  { id: 1, name: 'John', age: 30 },
  { id: 2, name: 'Jane', age: 25 },
  { id: 3, name: 'Jim', age: 40 },
];

export default function DataGridDemo() {
  const [selectedRows, setSelectedRows] = React.useState([rows[0].id]);

  const handleSelectionModelChange = (selectionModel) => {
    setSelectedRows(selectionModel);
  };

  return (
    <div style={{ height: 300, width: '100%' }}>
      <DataGrid
        rows={rows}
        columns={columns}
        rowSelection={{
          selectedRows,
          onChange: handleSelectionModelChange,
        }}
        checkboxSelection
      />
    </div>
  );
}