terapyon commited on
Commit
b1bf02b
·
1 Parent(s): bee398c

length, audio を追加した

Browse files
Files changed (1) hide show
  1. 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