pandas的DataFrame用法

用来生成DataFrame数据

1.说明：

class pandas.DataFrame(data=None, index=None, columns=None, dtype=None, copy=False)

Two-dimensional size-mutable, potentially heterogeneous tabular data structure with labeled axes (rows and columns). Arithmetic operations align on both row and column labels. Can be thought of as a dict-like container for Series objects. The primary pandas data structure.

Parameters:

data : numpy ndarray (structured or homogeneous), dict, or DataFrame Dict can contain Series, arrays, constants, or list-like objects Changed in version 0.23.0: If data is a dict, argument order is maintained for Python 3.6 and later. index : Index or array-like Index to use for resulting frame. Will default to RangeIndex if no indexing information part of input data and no index provided columns : Index or array-like Column labels to use for resulting frame. Will default to RangeIndex (0, 1, 2, …, n) if no column labels are provided dtype : dtype, default None Data type to force. Only a single dtype is allowed. If None, infer copy : boolean, default False Copy data from inputs. Only affects DataFrame / 2d ndarray input

代码：

 import tensorflow
 import lightgbm as lgb
 import pandas as pd
 import numpy as np

 class Deng(object):
     def __init__(self):
         pass

     def main(self):
         temp = ['a', 'a', 'b', 'c', 'c']
         st = pd.Categorical(temp)
         print(st)
         # [a, a, b, c, c]
         # Categories(3, object): [a, b, c]

         # 遍历temp指出temp中每个字符所属类别的位置索引
         st2 = st.codes
         print(st2)
         # [0 0 1 2 2]

     def gen_data(self):
         df = pd.DataFrame(data=np.eye(3), columns=['c1', 'c2', 'c3'])
         print(df)

 if __name__ == '__main__':
     obj = Deng()
     obj.gen_data()

输出：

    c1   c2   c3
0  1.0  0.0  0.0
1  0.0  1.0  0.0
2  0.0  0.0  1.0