dataframe-textual 2.2.2__tar.gz → 2.11.0__tar.gz

{dataframe_textual-2.2.2 → dataframe_textual-2.11.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dataframe-textual
-Version: 2.2.2
+Version: 2.11.0
 Summary: Interactive terminal viewer/editor for tabular data
 Project-URL: Homepage, https://github.com/need47/dataframe-textual
 Project-URL: Repository, https://github.com/need47/dataframe-textual.git
@@ -39,7 +39,7 @@ Description-Content-Type: text/markdown
 
 # DataFrame Textual
 
-A powerful, interactive terminal-based viewer/editor for CSV/TSV/Excel/Parquet/JSON/NDJSON built with Python, [Polars](https://pola.rs/), and [Textual](https://textual.textualize.io/). Inspired by [VisiData](https://www.visidata.org/), this tool provides smooth keyboard navigation, data manipulation, and a clean interface for exploring tabular data directly in terminal with multi-tab support for multiple files!
+A powerful, interactive terminal-based viewer/editor for CSV/TSV/PSV/Excel/Parquet/JSON/NDJSON built with Python, [Polars](https://pola.rs/), and [Textual](https://textual.textualize.io/). Inspired by [VisiData](https://www.visidata.org/), this tool provides smooth keyboard navigation, data manipulation, and a clean interface for exploring tabular data directly in terminal with multi-tab support for multiple files!
 
 ![Screenshot](https://raw.githubusercontent.com/need47/dataframe-textual/refs/heads/main/screenshot.png)
 
@@ -167,28 +167,32 @@ When multiple files are opened:
 ## Command Line Options
 
 ```
-usage: dv [-h] [-f {csv,excel,tsv,parquet,json,ndjson}] [-H] [-I] [-E] [-c COMMENT_PREFIX] [-q QUOTE_CHAR] [-l SKIP_LINES] [-a SKIP_ROWS_AFTER_HEADER] [-n NULL [NULL ...]] [files ...]
+usage: dv [-h] [-V] [-f {csv,json,xlsx,xls,ndjson,psv,parquet,tsv}] [-H] [-I] [-t] [-E] [-c [COMMENT_PREFIX]] [-q [QUOTE_CHAR]] [-l SKIP_LINES] [-a SKIP_ROWS_AFTER_HEADER] [-n NULL [NULL ...]] [files ...]
 
-Interactive terminal based viewer/editor for tabular data (e.g., CSV/TSV/Excel).
+Interactive terminal based viewer/editor for tabular data (e.g., CSV/Excel).
 
 positional arguments:
-  files                 Input files (or read from stdin)
+  files                 Files to view (or read from stdin)
 
 options:
   -h, --help            show this help message and exit
-  -f, --format {csv,excel,tsv,parquet,json,ndjson}
-                        Specify the format of the input files
-  -H, --no-header       Specify that input files have no header row
-  -I, --no-inference   Do not infer data types when reading CSV/TSV
+  -V, --version         show program's version number and exit
+  -f, --format {csv,json,xlsx,xls,ndjson,psv,parquet,tsv}
+                        Specify the format of the input files (csv, tsv etc.)
+  -H, --no-header       Specify that input files have no header row when reading CSV/TSV
+  -I, --no-inference    Do not infer data types when reading CSV/TSV
+  -t, --truncate-ragged-lines
+                        Truncate ragged lines when reading CSV/TSV
   -E, --ignore-errors   Ignore errors when reading CSV/TSV
-  -c, --comment-prefix COMMENT_PREFIX
-                        Comment lines are skipped when reading CSV/TSV (default: skip none)
-  -q, --quote-char QUOTE_CHAR
-                        Quote character for reading CSV/TSV (default: "; use -q without argument value to disable)
-  -l, --skip-lines SKIP_LINES
-                        Skip lines when reading CSV/TSV (default: 0)
-  -a, --skip-rows-after-header SKIP_ROWS_AFTER_HEADER
-                        Skip rows after header when reading CSV/TSV (default: 0)
+  -c, --comment-prefix [PREFIX]
+                        Comment lines starting with `PREFIX` are skipped when reading CSV/TSV
+  -q, --quote-char [C]
+                        Use `C` as quote character for reading CSV/TSV
+  -L, --skip-lines N
+                        Skip first N lines when reading CSV/TSV
+  -A, --skip-rows-after-header N
+                        Skip N rows after header when reading CSV/TSV
+  -N, --n-rows N        Stop after reading N rows from CSV/TSV
   -n, --null NULL [NULL ...]
                         Values to interpret as null values when reading CSV/TSV
 ```
@@ -243,11 +247,11 @@ zcat compressed_data.csv.gz | dv -f csv
 | `<` | Move to previous tab |
 | `b` | Cycle through tabs |
 | `B` | Toggle tab bar visibility |
-| `q` | Close current tab (prompts to save unsaved changes) |
-| `Q` | Close all tabs and app (prompts to save unsaved changes) |
+| `q` | Quit current tab (prompts to save unsaved changes) |
+| `Q` | Quit all tabs and app (prompts to save unsaved changes) |
 | `Ctrl+Q` | Force to quit app (regardless of unsaved changes) |
 | `Ctrl+T` | Save current tab to file |
-| `Ctrl+A` | Save all tabs to a Excel file |
+| `Ctrl+S` | Save all tabs to file |
 | `w` | Save current tab to file (overwrite without prompt) |
 | `W` | Save all tabs to file (overwrite without prompt) |
 | `Ctrl+D` | Duplicate current tab |
@@ -302,15 +306,17 @@ zcat compressed_data.csv.gz | dv -f csv
 | `_` (underscore) | Toggle column full width |
 | `z` | Freeze rows and columns |
 | `,` | Toggle thousand separator for numeric display |
+| `&` | Set current row as the new header row | 
 | `h` | Hide current column |
-| `H` | Show all hidden rows/columns |
+| `H` | Show all hidden columns |
 
-#### Data Editing
+#### Editing
 
 | Key | Action |
 |-----|--------|
 | `Double-click` | Edit cell or rename column header |
-| `delete` | Clear current cell (set to NULL) |
+| `Delete` | Clear current cell (set to NULL) |
+| `Shift+Delete` | Clear current column (set matching cells to NULL) |
 | `e` | Edit current cell (respects data type) |
 | `E` | Edit entire column with value/expression |
 | `a` | Add empty column after current |
@@ -352,16 +358,10 @@ zcat compressed_data.csv.gz | dv -f csv
 | Key | Action |
 |-----|--------|
 | `"` (quote) | Filter selected rows (others removed) |
+| `.` | View rows with non-null values in current column (others hidden) |
 | `v` | View selected rows (others hidden) |
 | `V` | View selected by expression (others hidden) |
 
-#### SQL Interface
-
-| Key | Action |
-|-----|--------|
-| `l` | Simple SQL interface (select columns & where clause) |
-| `L` | Advanced SQL interface (full SQL query with syntax highlight) |
-
 #### Sorting (supporting multiple columns)
 
 | Key | Action |
@@ -387,14 +387,20 @@ zcat compressed_data.csv.gz | dv -f csv
 | `!` | Cast current column to boolean |
 | `$` | Cast current column to string |
 
-#### Copy & Save
+#### Copy
 
 | Key | Action |
 |-----|--------|
 | `c` | Copy current cell to clipboard |
 | `Ctrl+C` | Copy column to clipboard |
 | `Ctrl+R` | Copy row to clipboard (tab-separated) |
-| `Ctrl+S` | Save to file |
+
+#### SQL Interface
+
+| Key | Action |
+|-----|--------|
+| `l` | Simple SQL interface (select columns & where clause) |
+| `L` | Advanced SQL interface (full SQL query with syntax highlight) |
 
 ## Features in Detail
 
@@ -417,6 +423,8 @@ Useful for examining wide datasets where columns don't fit well on screen.
 - Press `"` to **filter** all rows containing the selected column value (others removed)
 - Press `{` to move to the previous row
 - Press `}` to move to the next row
+- Press `F` to show the frequency table for the selected column
+- Press `s` to show the statistics table for the selected column
 - Press `q` or `Escape` to close the modal
 
 ### 3. Row Selection
@@ -442,7 +450,6 @@ These options work with plain text searches. Use Polars regex patterns in expres
 **Quick Tips:**
 - Search results highlight matching rows in **red**
 - Use expression for advanced selection (e.g., $attack > $defense)
-- Multiple searches **accumulate** - each new search adds to the selections or matches
 - Type-aware matching automatically converts values. Resort to string comparison if conversion fails
 - Use `u` to undo any search or filter
 
@@ -492,7 +499,7 @@ Both operations show selected rows but with fundamentally different effects:
 **When to use View** (`v` or `V`):
 - Exploring or analyzing data safely
 - Switching between different perspectives
-- Press `H` to restore hidden rows (and hidden columns)
+- Press `q` to return to main table
 
 **When to use Filter** (`"`):
 - Cleaning data (removing unwanted rows)
@@ -582,6 +589,10 @@ View quick metadata about your dataframe and columns to understand their structu
   - **Name** - Column name
   - **Type** - Data type (e.g., Int64, String, Float64, Boolean)
 
+**In the Column Metadata Table**
+- Press `F` to show the frequency table for the selected column
+- Press `s` to show the statistics table for the selected column
+
 **In Metadata Modals**:
 - Press `q` or `Escape` to close
 
@@ -621,7 +632,7 @@ This is useful for:
 - Quick statistical summaries without external tools
 - Comparing statistics across columns
 
-### 11. Data Editing
+### 11. Editing
 
 **Edit Cell** (`e` or **Double-click**):
 - Opens modal for editing current cell
@@ -667,7 +678,7 @@ This is useful for:
 
 **Hide/Show Columns** (`h` / `H`):
 - `h` - Temporarily hide current column (data preserved)
-- `H` - Restore all hidden columns and rows
+- `H` - Restore all hidden columns
 
 ### 12. Column & Row Reordering
 
@@ -696,7 +707,7 @@ Press `,` to toggle thousand separator formatting for numeric data:
 
 ### 15. Save File
 
-Press `Ctrl+S` to save filtered, edited, or sorted data back to file
+Press `Ctrl+S` to save filtered, edited, or sorted data back to file. The output format is automatically determined by the file extension, making it easy to convert between different formats (e.g., CSV to TSV).
 
 ### 16. Undo/Redo/Reset
 
@@ -816,7 +827,7 @@ Manage multiple files and dataframes simultaneously with tabs.
 - **`Double-click`** - Rename the tab
 - **`Ctrl+D`** - Duplicate current tab (creates a copy with same data and state)
 - **`Ctrl+T`** - Save current tab to file
-- **`Ctrl+A`** - Save all tabs in a single Excel file
+- **`Ctrl+S`** - Save all tabs to file
 - **`w`** - Save current tab to file (overwrite without prompt)
 - **`W`** - Save all tabs to file (overwrite without prompt)
 - **`q`** - Close current tab (closes tab, prompts to save if unsaved changes)

{dataframe_textual-2.2.2 → dataframe_textual-2.11.0}/README.md

@@ -1,6 +1,6 @@
 # DataFrame Textual
 
-A powerful, interactive terminal-based viewer/editor for CSV/TSV/Excel/Parquet/JSON/NDJSON built with Python, [Polars](https://pola.rs/), and [Textual](https://textual.textualize.io/). Inspired by [VisiData](https://www.visidata.org/), this tool provides smooth keyboard navigation, data manipulation, and a clean interface for exploring tabular data directly in terminal with multi-tab support for multiple files!
+A powerful, interactive terminal-based viewer/editor for CSV/TSV/PSV/Excel/Parquet/JSON/NDJSON built with Python, [Polars](https://pola.rs/), and [Textual](https://textual.textualize.io/). Inspired by [VisiData](https://www.visidata.org/), this tool provides smooth keyboard navigation, data manipulation, and a clean interface for exploring tabular data directly in terminal with multi-tab support for multiple files!
 
 ![Screenshot](https://raw.githubusercontent.com/need47/dataframe-textual/refs/heads/main/screenshot.png)
 
@@ -128,28 +128,32 @@ When multiple files are opened:
 ## Command Line Options
 
 ```
-usage: dv [-h] [-f {csv,excel,tsv,parquet,json,ndjson}] [-H] [-I] [-E] [-c COMMENT_PREFIX] [-q QUOTE_CHAR] [-l SKIP_LINES] [-a SKIP_ROWS_AFTER_HEADER] [-n NULL [NULL ...]] [files ...]
+usage: dv [-h] [-V] [-f {csv,json,xlsx,xls,ndjson,psv,parquet,tsv}] [-H] [-I] [-t] [-E] [-c [COMMENT_PREFIX]] [-q [QUOTE_CHAR]] [-l SKIP_LINES] [-a SKIP_ROWS_AFTER_HEADER] [-n NULL [NULL ...]] [files ...]
 
-Interactive terminal based viewer/editor for tabular data (e.g., CSV/TSV/Excel).
+Interactive terminal based viewer/editor for tabular data (e.g., CSV/Excel).
 
 positional arguments:
-  files                 Input files (or read from stdin)
+  files                 Files to view (or read from stdin)
 
 options:
   -h, --help            show this help message and exit
-  -f, --format {csv,excel,tsv,parquet,json,ndjson}
-                        Specify the format of the input files
-  -H, --no-header       Specify that input files have no header row
-  -I, --no-inference   Do not infer data types when reading CSV/TSV
+  -V, --version         show program's version number and exit
+  -f, --format {csv,json,xlsx,xls,ndjson,psv,parquet,tsv}
+                        Specify the format of the input files (csv, tsv etc.)
+  -H, --no-header       Specify that input files have no header row when reading CSV/TSV
+  -I, --no-inference    Do not infer data types when reading CSV/TSV
+  -t, --truncate-ragged-lines
+                        Truncate ragged lines when reading CSV/TSV
   -E, --ignore-errors   Ignore errors when reading CSV/TSV
-  -c, --comment-prefix COMMENT_PREFIX
-                        Comment lines are skipped when reading CSV/TSV (default: skip none)
-  -q, --quote-char QUOTE_CHAR
-                        Quote character for reading CSV/TSV (default: "; use -q without argument value to disable)
-  -l, --skip-lines SKIP_LINES
-                        Skip lines when reading CSV/TSV (default: 0)
-  -a, --skip-rows-after-header SKIP_ROWS_AFTER_HEADER
-                        Skip rows after header when reading CSV/TSV (default: 0)
+  -c, --comment-prefix [PREFIX]
+                        Comment lines starting with `PREFIX` are skipped when reading CSV/TSV
+  -q, --quote-char [C]
+                        Use `C` as quote character for reading CSV/TSV
+  -L, --skip-lines N
+                        Skip first N lines when reading CSV/TSV
+  -A, --skip-rows-after-header N
+                        Skip N rows after header when reading CSV/TSV
+  -N, --n-rows N        Stop after reading N rows from CSV/TSV
   -n, --null NULL [NULL ...]
                         Values to interpret as null values when reading CSV/TSV
 ```
@@ -204,11 +208,11 @@ zcat compressed_data.csv.gz | dv -f csv
 | `<` | Move to previous tab |
 | `b` | Cycle through tabs |
 | `B` | Toggle tab bar visibility |
-| `q` | Close current tab (prompts to save unsaved changes) |
-| `Q` | Close all tabs and app (prompts to save unsaved changes) |
+| `q` | Quit current tab (prompts to save unsaved changes) |
+| `Q` | Quit all tabs and app (prompts to save unsaved changes) |
 | `Ctrl+Q` | Force to quit app (regardless of unsaved changes) |
 | `Ctrl+T` | Save current tab to file |
-| `Ctrl+A` | Save all tabs to a Excel file |
+| `Ctrl+S` | Save all tabs to file |
 | `w` | Save current tab to file (overwrite without prompt) |
 | `W` | Save all tabs to file (overwrite without prompt) |
 | `Ctrl+D` | Duplicate current tab |
@@ -263,15 +267,17 @@ zcat compressed_data.csv.gz | dv -f csv
 | `_` (underscore) | Toggle column full width |
 | `z` | Freeze rows and columns |
 | `,` | Toggle thousand separator for numeric display |
+| `&` | Set current row as the new header row | 
 | `h` | Hide current column |
-| `H` | Show all hidden rows/columns |
+| `H` | Show all hidden columns |
 
-#### Data Editing
+#### Editing
 
 | Key | Action |
 |-----|--------|
 | `Double-click` | Edit cell or rename column header |
-| `delete` | Clear current cell (set to NULL) |
+| `Delete` | Clear current cell (set to NULL) |
+| `Shift+Delete` | Clear current column (set matching cells to NULL) |
 | `e` | Edit current cell (respects data type) |
 | `E` | Edit entire column with value/expression |
 | `a` | Add empty column after current |
@@ -313,16 +319,10 @@ zcat compressed_data.csv.gz | dv -f csv
 | Key | Action |
 |-----|--------|
 | `"` (quote) | Filter selected rows (others removed) |
+| `.` | View rows with non-null values in current column (others hidden) |
 | `v` | View selected rows (others hidden) |
 | `V` | View selected by expression (others hidden) |
 
-#### SQL Interface
-
-| Key | Action |
-|-----|--------|
-| `l` | Simple SQL interface (select columns & where clause) |
-| `L` | Advanced SQL interface (full SQL query with syntax highlight) |
-
 #### Sorting (supporting multiple columns)
 
 | Key | Action |
@@ -348,14 +348,20 @@ zcat compressed_data.csv.gz | dv -f csv
 | `!` | Cast current column to boolean |
 | `$` | Cast current column to string |
 
-#### Copy & Save
+#### Copy
 
 | Key | Action |
 |-----|--------|
 | `c` | Copy current cell to clipboard |
 | `Ctrl+C` | Copy column to clipboard |
 | `Ctrl+R` | Copy row to clipboard (tab-separated) |
-| `Ctrl+S` | Save to file |
+
+#### SQL Interface
+
+| Key | Action |
+|-----|--------|
+| `l` | Simple SQL interface (select columns & where clause) |
+| `L` | Advanced SQL interface (full SQL query with syntax highlight) |
 
 ## Features in Detail
 
@@ -378,6 +384,8 @@ Useful for examining wide datasets where columns don't fit well on screen.
 - Press `"` to **filter** all rows containing the selected column value (others removed)
 - Press `{` to move to the previous row
 - Press `}` to move to the next row
+- Press `F` to show the frequency table for the selected column
+- Press `s` to show the statistics table for the selected column
 - Press `q` or `Escape` to close the modal
 
 ### 3. Row Selection
@@ -403,7 +411,6 @@ These options work with plain text searches. Use Polars regex patterns in expres
 **Quick Tips:**
 - Search results highlight matching rows in **red**
 - Use expression for advanced selection (e.g., $attack > $defense)
-- Multiple searches **accumulate** - each new search adds to the selections or matches
 - Type-aware matching automatically converts values. Resort to string comparison if conversion fails
 - Use `u` to undo any search or filter
 
@@ -453,7 +460,7 @@ Both operations show selected rows but with fundamentally different effects:
 **When to use View** (`v` or `V`):
 - Exploring or analyzing data safely
 - Switching between different perspectives
-- Press `H` to restore hidden rows (and hidden columns)
+- Press `q` to return to main table
 
 **When to use Filter** (`"`):
 - Cleaning data (removing unwanted rows)
@@ -543,6 +550,10 @@ View quick metadata about your dataframe and columns to understand their structu
   - **Name** - Column name
   - **Type** - Data type (e.g., Int64, String, Float64, Boolean)
 
+**In the Column Metadata Table**
+- Press `F` to show the frequency table for the selected column
+- Press `s` to show the statistics table for the selected column
+
 **In Metadata Modals**:
 - Press `q` or `Escape` to close
 
@@ -582,7 +593,7 @@ This is useful for:
 - Quick statistical summaries without external tools
 - Comparing statistics across columns
 
-### 11. Data Editing
+### 11. Editing
 
 **Edit Cell** (`e` or **Double-click**):
 - Opens modal for editing current cell
@@ -628,7 +639,7 @@ This is useful for:
 
 **Hide/Show Columns** (`h` / `H`):
 - `h` - Temporarily hide current column (data preserved)
-- `H` - Restore all hidden columns and rows
+- `H` - Restore all hidden columns
 
 ### 12. Column & Row Reordering
 
@@ -657,7 +668,7 @@ Press `,` to toggle thousand separator formatting for numeric data:
 
 ### 15. Save File
 
-Press `Ctrl+S` to save filtered, edited, or sorted data back to file
+Press `Ctrl+S` to save filtered, edited, or sorted data back to file. The output format is automatically determined by the file extension, making it easy to convert between different formats (e.g., CSV to TSV).
 
 ### 16. Undo/Redo/Reset
 
@@ -777,7 +788,7 @@ Manage multiple files and dataframes simultaneously with tabs.
 - **`Double-click`** - Rename the tab
 - **`Ctrl+D`** - Duplicate current tab (creates a copy with same data and state)
 - **`Ctrl+T`** - Save current tab to file
-- **`Ctrl+A`** - Save all tabs in a single Excel file
+- **`Ctrl+S`** - Save all tabs to file
 - **`w`** - Save current tab to file (overwrite without prompt)
 - **`W`** - Save all tabs to file (overwrite without prompt)
 - **`q`** - Close current tab (closes tab, prompts to save if unsaved changes)

{dataframe_textual-2.2.2 → dataframe_textual-2.11.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "dataframe-textual"
-version = "2.2.2"
+version = "2.11.0"
 description = "Interactive terminal viewer/editor for tabular data"
 readme = "README.md"
 requires-python = ">=3.11"
@@ -56,11 +56,13 @@ dev = [
 packages = ["src/dataframe_textual"]
 exclude = [
     "*.png",
+    "*.gz",
 ]
 
 [tool.hatch.build.targets.sdist]
 exclude = [
     "*.png",
+    "*.gz",
 ]
 
 [dependency-groups]

{dataframe_textual-2.2.2 → dataframe_textual-2.11.0}/src/dataframe_textual/__main__.py

@@ -51,15 +51,34 @@ def cli() -> argparse.Namespace:
     )
     parser.add_argument("-E", "--ignore-errors", action="store_true", help="Ignore errors when reading CSV/TSV")
     parser.add_argument(
-        "-c", "--comment-prefix", nargs="?", const="#", help="Comment lines are skipped when reading CSV/TSV"
+        "-c",
+        "--comment-prefix",
+        metavar="PREFIX",
+        nargs="?",
+        const="#",
+        help="Comment lines starting with `PREFIX` are skipped when reading CSV/TSV",
     )
     parser.add_argument(
-        "-q", "--quote-char", nargs="?", const=None, default='"', help="Quote character for reading CSV/TSV"
+        "-q",
+        "--quote-char",
+        metavar="C",
+        nargs="?",
+        const=None,
+        default='"',
+        help="Use `C` as quote character for reading CSV/TSV",
     )
-    parser.add_argument("-l", "--skip-lines", type=int, default=0, help="Skip lines when reading CSV/TSV")
     parser.add_argument(
-        "-a", "--skip-rows-after-header", type=int, default=0, help="Skip rows after header when reading CSV/TSV"
+        "-L", "--skip-lines", metavar="N", type=int, default=0, help="Skip first N lines when reading CSV/TSV"
     )
+    parser.add_argument(
+        "-A",
+        "--skip-rows-after-header",
+        metavar="N",
+        type=int,
+        default=0,
+        help="Skip N rows after header when reading CSV/TSV",
+    )
+    parser.add_argument("-N", "--n-rows", metavar="N", type=int, help="Stop after reading N rows from CSV/TSV")
     parser.add_argument("-n", "--null", nargs="+", help="Values to interpret as null values when reading CSV/TSV")
 
     args = parser.parse_args()
@@ -98,6 +117,7 @@ def main() -> None:
         null_values=args.null,
         ignore_errors=args.ignore_errors,
         truncate_ragged_lines=args.truncate_ragged_lines,
+        n_rows=args.n_rows,
     )
     app = DataFrameViewer(*sources)
     app.run()

{dataframe_textual-2.2.2 → dataframe_textual-2.11.0}/src/dataframe_textual/common.py

@@ -12,7 +12,7 @@ import polars as pl
 from rich.text import Text
 
 # Supported file formats
-SUPPORTED_FORMATS = {"tsv", "tab", "csv", "excel", "xlsx", "xls", "parquet", "json", "ndjson"}
+SUPPORTED_FORMATS = ["tsv", "csv", "psv", "xlsx", "xls", "parquet", "json", "ndjson"]
 
 
 # Boolean string mappings
@@ -485,6 +485,7 @@ def load_dataframe(
     null_values: list[str] | None = None,
     ignore_errors: bool = False,
     truncate_ragged_lines: bool = False,
+    n_rows: int | None = None,
 ) -> list[Source]:
     """Load DataFrames from file specifications.
 
@@ -502,6 +503,8 @@ def load_dataframe(
         skip_rows_after_header: Number of rows to skip after header. Defaults to 0.
         null_values: List of values to interpret as null when reading CSV/TSV files. Defaults to None.
         ignore_errors: Whether to ignore errors when reading CSV/TSV files. Defaults to False.
+        truncate_ragged_lines: Whether to truncate ragged lines when reading CSV/TSV files. Defaults to False.
+        n_rows: Number of rows to read from CSV/TSV files. Defaults to None (read all rows).
 
     Returns:
         List of `Source` objects.
@@ -529,6 +532,7 @@ def load_dataframe(
             ext = Path(filename).suffix.lower()
             if ext == ".gz":
                 ext = Path(filename).with_suffix("").suffix.lower()
+
             fmt = ext.removeprefix(".")
 
             # Default to TSV
@@ -550,6 +554,7 @@ def load_dataframe(
                 null_values=null_values,
                 ignore_errors=ignore_errors,
                 truncate_ragged_lines=truncate_ragged_lines,
+                n_rows=n_rows,
             )
         )
 
@@ -634,6 +639,7 @@ def load_file(
     null_values: list[str] | None = None,
     ignore_errors: bool = False,
     truncate_ragged_lines: bool = False,
+    n_rows: int | None = None,
 ) -> list[Source]:
     """Load a single file.
 
@@ -659,6 +665,8 @@ def load_file(
         schema_overrides: Optional dictionary of column name to Polars data type to override inferred schema.
         null_values: List of values to interpret as null when reading CSV/TSV files. Defaults to None.
         ignore_errors: Whether to ignore errors when reading CSV/TSV files.
+        truncate_ragged_lines: Whether to truncate ragged lines when reading CSV/TSV files. Defaults to False.
+        n_rows: Number of rows to read from CSV/TSV files. Defaults to None (read all rows).
 
     Returns:
         List of `Source` objects.
@@ -668,17 +676,11 @@ def load_file(
     filename = f"stdin.{file_format}" if isinstance(source, StringIO) else source
     filepath = Path(filename)
 
-    if not file_format:
-        ext = filepath.suffix.lower()
-        if ext == ".gz":
-            ext = Path(filename).with_suffix("").suffix.lower()
-        file_format = ext.removeprefix(".")
-
     # Load based on file format
-    if file_format in ("csv", "tsv"):
+    if file_format in ("csv", "tsv", "psv"):
         lf = pl.scan_csv(
             source,
-            separator="\t" if file_format == "tsv" else ",",
+            separator="\t" if file_format == "tsv" else ("|" if file_format == "psv" else ","),
             has_header=has_header,
             infer_schema=infer_schema,
             comment_prefix=comment_prefix,
@@ -689,9 +691,10 @@ def load_file(
             null_values=null_values,
             ignore_errors=ignore_errors,
             truncate_ragged_lines=truncate_ragged_lines,
+            n_rows=n_rows,
         )
         data.append(Source(lf, filename, filepath.stem))
-    elif file_format in ("xlsx", "xls", "excel"):
+    elif file_format in ("xlsx", "xls"):
         if first_sheet:
             # Read only the first sheet for multiple files
             lf = pl.read_excel(source).lazy()
@@ -745,6 +748,8 @@ def load_file(
             schema_overrides=schema_overrides,
             null_values=null_values,
             ignore_errors=ignore_errors,
+            truncate_ragged_lines=truncate_ragged_lines,
+            n_rows=n_rows,
         )
 
     return data