python ndarray与pandas series相互转换，ndarray与dataframe相互转换

╰半橙微兮° 2021-11-09 18:24 1841阅读 0赞

用python做科学计算时，经常需要类型转换，以下是常用类型转换

### 一、ndarray 转换为 series ###

1、如果ndarray是二维数组，如下

array([[1],
           [2],
           [3]])

需要通过map结合lamdba

import numpy as np
    import pandas as pd
    
    data = np.array([1, 2, 3]).reshape(3, 1)
    data_list = map(lambda x: x[0], data)
    ser = pd.Series(data_list)

2、如果ndarray是一维数组，如下

array(\[1, 2, 3\])

则通过data.tolist()即可

import numpy as np
    import pandas as pd
    
    data = np.array([1, 2, 3])
    ser = pd.Series(data.tolist())

### 二、series转换为ndarray ###

通过Series.values实现series转换为ndarray

import pandas as pd
    
    data = [['2019/08/01', 10],
            ['2019/08/01', 11]]
    result = pd.DataFrame(data, columns=['ds', 'val'])
    result['val'].values
    
    data2 = pd.Series([1, 2, 3])
    data2.values

### 三、ndarray转换为dataframe ###

1、直接通过pd.DataFrame转换

import numpy as np
    import pandas as pd
    
    data = np.array([['2019/08/02', 'zhansan', 1], ['2019/08/03', 'lisi', 2], ['2019/08/04', 'wangwu', 3]])
    df = pd.DataFrame(data)

打印df为

![20190802000500312.png][]

2、指定索引、数据、列名例子

import numpy as np
    import pandas as pd
    
    data = np.array([['', 'Col1', 'Col2'], ['Row1', 1, 2], ['Row2', 3, 4]])
    df = pd.DataFrame(data=data[1:, 1:],  # 从第2行开始并且第2列开始作为数据
                      index=data[1:, 0],  # 第1列做索引，从第2行开始
                      columns=data[0, 1:])  # 第1行作为列名，从第2列开始

![20190802001941218.png][]

### 四、dataframe转换为ndarray ###

1、通过values方法，实现dataframe转换为ndarray

import pandas as pd
    
    data = [['2019/08/01', 10],
            ['2019/08/01', 11]]
    result = pd.DataFrame(data, columns=['ds', 'val'])
    result.values

dataframe数据![20190802091920228.png][]

转换后的是数据![20190802091933877.png][]

2、通过切片，实现某一行或者某一列转换为ndarray

import pandas as pd
    
    data = [['2019/08/01', 10],
            ['2019/08/01', 11]]
    result = pd.DataFrame(data, columns=['ds', 'val'])
    rs = result.values
    print(rs[:, 0])
    print(rs[0, :])

rs[:, 0] 逗号前面表示：行都保留，逗号后面0表示保留第一列，结果为['2019/08/01' '2019/08/01']

rs[0, :]表示保留第一行，列都保留，结果为['2019/08/01' 10]

[20190802000500312.png]: /images/20211109/d9c979c59d8e4c409fb00f8b299332b3.png
[20190802001941218.png]: /images/20211109/f4ca4125d39c481d9acd1b3819754c2f.png
[20190802091920228.png]: /images/20211109/6d14acc45a4946feaa5bdce4b38f89cb.png
[20190802091933877.png]: /images/20211109/595bb10fff7b4963b10c5ca6a0cf2b4a.png