Python 帶 if 語句的單行 for 循環

Question

我有一個 function 用於我的 Pandas dataframe 以返回每列的長度：

def get_col_widths(dataframe)
    return [max([len(str(s)) for s in dataframe[col].values] + [len(col)]) for col in dataframe.columns]

但是，我有一列日期（類型為numpy.datetime64 ），因此當您執行len(str(s))時，它會返回 29 而不是預期的 10 （ MM/DD/YYYY ）。

如何合並 if 來測試isinstance(s, np.datetime64)是否返回 10 而不是len(str(s)) ？

Answer 1

像這樣：

def get_col_widths(dataframe):
    return [max([len(str(s)) if not isinstance(s, np.datetime64) else 10 for s in dataframe[col].values] + [len(col)]) for col in dataframe.columns]

Answer 2

您只需在for s之前滑過條件，如下所示：

def get_col_widths(dataframe):
     return [max([(len(str(s)) if not isinstance(s, np.datetime64) else 10) for s in dataframe[col].values] + [len(col)]) for col in dataframe.columns]

但我建議不要使用單線。 無需壓縮所有內容。 可讀代碼更好。

Answer 3

我們這里不需要 for 循環

c=col

df[c].astype(str).str.len().max()