DAPyDataFrame.h

#ifndef DAPYDATAFRAME_H
#define DAPYDATAFRAME_H
#include "DAPyBindQtGlobal.h"
#include "DAPyObjectWrapper.h"
#include "DAPyIndex.h"
#include "DAPybind11InQt.h"
#include "DAPySeries.h"
#include <QVariant>
#include <QDebug>
#include <QList>
namespace DA
{
class DAPYBINDQT_API DAPyDataFrame : public DAPyObjectWrapper
{
public:
    DAPyDataFrame();
    DAPyDataFrame(const DAPyDataFrame& df);
    DAPyDataFrame(const DAPyObjectWrapper& df);
    DAPyDataFrame(DAPyDataFrame&& df);
    DAPyDataFrame(const pybind11::object& obj);
    DAPyDataFrame(pybind11::object&& obj);
    ~DAPyDataFrame();
    // 获取列
    // 注意pandas.dataframe的列名是可以任意类型，如果列名是数字，那么熟通过数字来获取而不是用字符串，
 
    DAPySeries operator[](const QString& n) const;
    DAPySeries operator[](size_t n) const;
    DAPyDataFrame& operator=(const pybind11::object& obj);
    DAPyDataFrame& operator=(const DAPyDataFrame& obj);
    DAPyDataFrame& operator=(const DAPyObjectWrapper& obj);
    DAPyDataFrame& operator=(pybind11::object&& obj);
    DAPyDataFrame& operator=(DAPyDataFrame&& obj);
    DAPyDataFrame& operator=(DAPyObjectWrapper&& obj);
    // 重写小于操作，用于std::map容器，注意，这个小于仅仅比较指针的大小，没有任何意义
    bool operator<(const DAPyDataFrame& other) const;
 
public:
    // DataFrame.columns
    QList< QString > columns() const;
    QString columnName(std::size_t i) const;
    bool columns(std::size_t i, const QString& name);
    bool columns(const QList< QString >& cols);
    // DataFrame.empty
    bool empty() const;
    // DataFrame.shape
    std::pair< std::size_t, std::size_t > shape() const;
    // DataFrame.size
    std::size_t size() const;
    // DataFrame.iat
    QVariant iat(std::size_t r, std::size_t c) const;
    pybind11::object iatObj(std::size_t r, std::size_t c) const;
    bool iat(std::size_t r, std::size_t c, const QVariant& v);
    bool iat(std::size_t r, std::size_t c, const pybind11::object& v);
    // DataFrame.loc 获取一行
    DAPySeries iloc(std::size_t c) const;
    DAPySeries loc(const QString& n) const;
    bool drop(std::size_t index, int axis = 0);
    DAPyIndex index() const;
    // 获取某一列对应的dtype
    pybind11::dtype dtypes(std::size_t c) const;
    DAPyDataFrame describe() const;
    // 转换为csv
    bool to_csv(const QString& path, const QVariantHash& args = QVariantHash()) const noexcept;
    // 转换为pickle
    bool to_pickle(const QString& path, const QVariantHash& args = QVariantHash()) const noexcept;
    // 转换为parquet
    bool to_parquet(const QString& path, const QVariantHash& args = QVariantHash()) const noexcept;
    // 设置为索引，错误会抛出异常
    bool setIndex(const QString& colName, bool isDrop = true) noexcept;
 
protected:
    // 检测是否为dataframe，如果不是将会设置为none
    void checkObjectValid();
 
public:
    QString toString(std::size_t maxrow = 12) const;
    static bool isDataFrame(const pybind11::object& obj);
 
public:
    bool read_csv(const QString& path, const QVariantHash args = QVariantHash(), QString* why = nullptr);
};
}  // namespace DA
DAPYBINDQT_API QDebug operator<<(QDebug dbg, const DA::DAPyDataFrame& df);
 
Q_DECLARE_METATYPE(DA::DAPyDataFrame)
#endif  // DADATAFRAME_H