使用pandas保存python二维列表为csv文件
numpy与pandas库常见需求总结,后续待补充。
一,将二维列表保存为csv格式文件
- 将数组实例化为np.array对象,通过pandas.DataFrame保存为对象,使用其to_csv方法保存为csv文件
import pandas as pd
import numpy as np
li = [['name', 'age', 'addres'], ['john', '18', 'bj'], ['jack', '26', 'sh']]
array = np.array(li)
print(type(array))
print(array)
df = pd.DataFrame(array)
print(type(df))
print(df)
df.to_csv('data.csv')
执行结果
<class 'numpy.ndarray'>
[['name' 'age' 'addres']
['john' '18' 'bj']
['jack' '26' 'sh']]
<class 'pandas.core.frame.DataFrame'>
0 1 2
0 name age addres
1 john 18 bj
2 jack 26 sh