Spaces:
Running
Running
length, audio を追加した
Browse files- src/titles.py +3 -3
src/titles.py
CHANGED
@@ -16,7 +16,7 @@ def parse_csv(filename: str | Path) -> pd.DataFrame:
|
|
16 |
df = pd.read_csv(filename,
|
17 |
header=None,
|
18 |
)
|
19 |
-
df.columns = ["id", "day_import", "date_import", "title_import"]
|
20 |
df["datetime"] = pd.to_datetime(df["date_import"], format=" %d %b %Y %H:%M:%S %Z")
|
21 |
df["date"] = df.loc[:, "datetime"].apply(lambda x: parse_ja_date(x))
|
22 |
df["title"] = df.loc[:, "title_import"].astype("string")
|
@@ -25,9 +25,9 @@ def parse_csv(filename: str | Path) -> pd.DataFrame:
|
|
25 |
|
26 |
def ignore_columns(df: pd.DataFrame, limit: str | None) -> pd.DataFrame:
|
27 |
if limit is not None:
|
28 |
-
df_out = df.loc[df.loc[:, "datetime"] > limit, ["id", "date", "title"]]
|
29 |
else:
|
30 |
-
df_out = df.loc[:, ["id", "date", "title"]]
|
31 |
return df_out
|
32 |
|
33 |
|
|
|
16 |
df = pd.read_csv(filename,
|
17 |
header=None,
|
18 |
)
|
19 |
+
df.columns = ["id", "day_import", "date_import", "length", "audio", "title_import"]
|
20 |
df["datetime"] = pd.to_datetime(df["date_import"], format=" %d %b %Y %H:%M:%S %Z")
|
21 |
df["date"] = df.loc[:, "datetime"].apply(lambda x: parse_ja_date(x))
|
22 |
df["title"] = df.loc[:, "title_import"].astype("string")
|
|
|
25 |
|
26 |
def ignore_columns(df: pd.DataFrame, limit: str | None) -> pd.DataFrame:
|
27 |
if limit is not None:
|
28 |
+
df_out = df.loc[df.loc[:, "datetime"] > limit, ["id", "date", "length", "audio", "title"]]
|
29 |
else:
|
30 |
+
df_out = df.loc[:, ["id", "date", "length", "audio", "title"]]
|
31 |
return df_out
|
32 |
|
33 |
|