编写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')