我可以有條件地使用熊貓read_csv轉換器嗎？

Question

我想知道是否可以根據列的類型有條件地使用轉換器。例如，如果數據集由兩列構成，其中第一列可以是整數或時間戳，如果文件最終有一個時間戳列，則可以有條件地使用轉換器，並避免在列中包含整數時調用轉換器值。

這可以在read_csv之外完成，當數據已經加載但它可能會效率較低。

Answer 1

read_csv docs指出您可以使用date_parser參數傳遞自定義日期解析函數。

所以，你可以這樣做：

from datetime import datetime 
import pandas as pd 
from dateutil.parser import parse 


def parse_date_or_ts(data): 
    try: 
     return parse(data) 
    except ValueError: 
     # let's assume data is a utc timestamp now 
     return datetime.utcfromtimestamp(int(data)) 

df = pd.read_csv('data.csv', 
       parse_dates={'date': [0]}, # index 0 is our date/ts mixed column 
       date_parser=parse_date_or_ts)