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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dataframe-textual
-Version: 2.3.0
+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 |
@@ -304,14 +308,15 @@ zcat compressed_data.csv.gz | dv -f csv
 | `,` | 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 |
@@ -353,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 |
@@ -388,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
 
@@ -418,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
@@ -443,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
 
@@ -493,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)
@@ -583,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
 
@@ -622,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
@@ -668,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
 
@@ -817,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.3.0 → 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 |
@@ -265,14 +269,15 @@ zcat compressed_data.csv.gz | dv -f csv
 | `,` | 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 |
@@ -314,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 |
@@ -349,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
 
@@ -379,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
@@ -404,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
 
@@ -454,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)
@@ -544,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
 
@@ -583,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
@@ -629,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
 
@@ -778,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.3.0 → dataframe_textual-2.11.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "dataframe-textual"
-version = "2.3.0"
+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.3.0 → 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.3.0 → 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