Adds 'date_format' parameter to 'read_csv' (#650)

* Added test for date_format in read_csv

* Added date_format to read_csv

* date_format int col mapping and extra tests

---------

Co-authored-by: Sjors Lockhorst <[email protected]>

Author: SjorsLockhorst

pandas-stubs/io/parsers/readers.pyi

@@ -67,6 +67,7 @@ def read_csv(
     infer_datetime_format: bool = ...,
     keep_date_col: bool = ...,
     date_parser: Callable = ...,
+    date_format: str | Mapping[int | str, str] | None = ...,
     dayfirst: bool = ...,
     cache_dates: bool = ...,
     iterator: Literal[True],
@@ -126,6 +127,7 @@ def read_csv(
     infer_datetime_format: bool = ...,
     keep_date_col: bool = ...,
     date_parser: Callable = ...,
+    date_format: str | Mapping[int | str, str] | None = ...,
     dayfirst: bool = ...,
     cache_dates: bool = ...,
     iterator: bool = ...,
@@ -185,6 +187,7 @@ def read_csv(
     infer_datetime_format: bool = ...,
     keep_date_col: bool = ...,
     date_parser: Callable = ...,
+    date_format: str | Mapping[int | str, str] | None = ...,
     dayfirst: bool = ...,
     cache_dates: bool = ...,
     iterator: Literal[False] = ...,

tests/test_io.py

@@ -638,6 +638,21 @@ def test_types_read_csv() -> None:
         tfr4: TextFileReader = pd.read_csv(path, nrows=2, iterator=True)
         tfr4.close()
 
+    df_dates = pd.DataFrame(data={"col1": ["2023-03-15", "2023-04-20"]})
+
+    with ensure_clean() as path:
+        df_dates.to_csv(path)
+
+        df26: pd.DataFrame = pd.read_csv(
+            path, parse_dates=["col1"], date_format="%Y-%m-%d"
+        )
+        df27: pd.DataFrame = pd.read_csv(
+            path, parse_dates=["col1"], date_format={"col1": "%Y-%m-%d"}
+        )
+        df28: pd.DataFrame = pd.read_csv(
+            path, parse_dates=["col1"], date_format={1: "%Y-%m-%d"}
+        )
+
 
 def test_read_table():
     with ensure_clean() as path: