创建一个五行三列的名为df1的DataFrame数组,列名为 [states,years,pops],行名['one','two','three','four','five']; 给df1添加新列,列名为new_add,值为[7,4,5,8,2]。def create_dataframe(): ''' 返回值: df1: 一个DataFrame类型数据 '''#返回df1
时间: 2023-08-06 16:18:22 浏览: 219
pandas.DataFrame删除/选取含有特定数值的行或列实例
好的,可以按照如下代码实现create_dataframe函数:
```python
import pandas as pd
import numpy as np
def create_dataframe():
data = np.array([['one', 2000, 1.5],
['two', 2001, 1.7],
['three', 2002, 3.6],
['four', 2001, 2.4],
['five', 2002, 2.9]])
df1 = pd.DataFrame(data, columns=['states', 'years', 'pops'], index=['one', 'two', 'three', 'four', 'five'])
df1['new_add'] = [7, 4, 5, 8, 2]
return df1
```
这个函数的功能是创建一个5行3列的DataFrame数组,并添加一列新的数据。具体实现中,我们首先使用numpy库中的array函数创建一个名为data的数组,其中包含5行3列的数据,分别表示states、years和pops这3个列的数据。接着,我们使用pd.DataFrame函数将data数组转化为一个名为df1的DataFrame数组,其中列名为[states,years,pops],行名['one','two','three','four','five']。最后,我们使用df1['new_add'] = [7, 4, 5, 8, 2]来添加一列新的数据,列名为new_add,值为[7,4,5,8,2]。最后,我们将创建好的df1返回即可。
阅读全文