編寫CSV的Python函數

Question

def write_csv(mydf):
    nominal = mydf.name 
    csvname = nominal + ".csv"
    mydf.to_csv( csvname , encoding='utf-8', header='False')

當我調用該函數時，此方法不起作用：

write_csv(ulangp) 
Attribute Error:  'DataFrame' object has no attribute 'name'

假設我要傳遞數據幀，在函數中執行此操作的正確方法是什么？

Answer 1

您可以傳遞[n可選] name參數，如果數據框沒有預設名稱，則使用該參數：

def write_csv(mydf, name="default"): # Use any good default name here
    try:
        nominal = mydf.name 
    except AttributeError:
        nominal = name
    csvname = nominal + ".csv"
    mydf.to_csv( csvname , encoding='utf-8', header='False')