Can I use Pandas in Mojo?

[cjohnson318]

Can I use Pandas in Mojo? I tried this:

from PythonInterface import Python
let pd = Python.import_module("pandas")

d = Python.dict()
d['col1']=[1, 2]
d['col2']=[3, 4]

df = pd.DataFrame(d)

But I get an error:

error: Expression [1]:21:22: invalid call to '__call__': argument #1 cannot be converted from 'PythonDictionary' to 'PythonObject'
    df = pd.DataFrame(d)
         ~~~~~~~~~~~~^~~

/.modular/Kernels/mojo/Python/PythonObject.mojo:881:5: function declared here
    fn __call__(self, *args: PythonObject) raises -> PythonObject:
    ^

[lattner]

@stumpOS is there a reason Python.dict() returns a PythonDictionary instead of PythonObject?

[cjohnson318]

The answer is to cast the PythonDictionary to a PythonObject by using the pyObject attribute.

from PythonInterface import Python
let pd = Python.import_module("pandas")
d = Python.dict()
d['col1']=[1, 2]
d['col2']=[3, 4]
df = pd.DataFrame(d.pyObject)
print(df)

Which returns:

   col1  col2
0     1     3
1     2     4

[testkevinkim]

Pandas in mojo ran faster ?

[soraros]

@testkevinkim No. Mojo uses CPython to run python libraries.

[HananyTolba]

You could do this with the following implementation:

from python import Python

fn main() raises:
    let pd = Python.import_module("pandas")
    let np = Python.import_module("numpy")

    let data = np.array([1, 2, 3, 2, 3, 4, 4, 5, 6]).reshape(3,3).T
    let df =  pd.DataFrame(data, np.arange(3), ['cola', 'colb', 'colc'])
    print(df)

[mohanreddy7892]

### converting` csv file to excel file

data.csv 
Name, Age, Occupation, Country
John Doe, 28, Engineer, USA
Jane Smith, 34, Designer, UK
Alice Johnson, 29, Artist, Australia
Bob Brown, 22, Student, Canada
Charlie White, 31, Doctor, New Zealand
Diana Green, 40, Lawyer, India
Eve Black, 25, Photographer, France
Frank Gray, 37, Journalist, Germany
Grace Blue, 29, Architect, Italy
Henry Pink, 45, Chef, Spain

from python import Python
fn main() raises:
    let pd = Python.import_module("pandas")
    let openpyxl = Python.import_module("openpyxl")
    let csv_file = "/home/mohan/data.csv"
    let excel_file = "/home/mohan/data.xlsx"
    let df = pd.read_csv(csv_file)
    print(df)
    df.to_excel(excel_file, index=False, engine=openpyxl)
    print("CSV has been converted to Excel!")

error:Unhandled exception caught during execution: An error occurred in Python.
mojo: error: execution exited with a non-zero result: 1