arff2pandas

A bidirectional converter arff to pandas.DataFrame


Keywords
arff, pandas, python3
License
MIT
Install
pip install arff2pandas==1.0.1

Documentation

arff2pandas

A bidirectional converter arff to pandas

install

pip install arff2pandas

require

  • python 3.*
  • liac-arff
  • pandas

usage

import

from arff2pandas import a2p

load arff from file to pandas.DataFrame

with open('sample.arff') as f:
    df = a2p.load(f)
    print(df)

load arff from str to pandas.DataFrame

arff_str = '@RELATION feature\n\n@ATTRIBUTE power NUMERIC\n@ATTRIBUTE label {good,bad}\n\n@DATA\n0.3,good\n0.5,bad'
df = a2p.loads(arff_str)
print(df)

dump pandas.DataFrame to file

import pandas as pd
df = pd.DataFrame({
    'power@NUMERIC':[0.5,0.2],
    'label@{good,bad}':['good','bad']
})
with open('sample.arff','w') as f:
    a2p.dump(df,f)

dump pandas.DataFrame to str

import pandas as pd
df = pd.DataFrame({
    'power@NUMERIC':[0.5,0.2],
    'label@{good,bad}':['good','bad']
})
print(a2p.dumps(df))