Pandas and Python for investing with sentiment

Pandas Basics

Code written in Python 2, but you can either comment out all print statements, or convert them to functions and it will work fine.

import datetime
import pandas as pd
from pandas import DataFrame
import matplotlib.pyplot as plt
from matplotlib import style


## grabbing from yahoo finance ###
sp500 ='%5EGSPC', 
                                 start=datetime.datetime(2000, 10, 1), 
                                 end=datetime.datetime(2012, 1, 1))
print sp500.head()

# saving to a csv #

# reading from the csv #
df = pd.read_csv('sp500_ohlc.csv', index_col='Date', parse_dates=True)

print df.index

# again, just accessing a column.... #

ts = df['Close'][-10:]
print ts

# or a couple:
print df[['Open', 'Close']].head()

# can also create new columns:
df['H-L'] = df.High - df.Low
print df.head()

# can also delete:
del df['H-L']
print df.head()

#So besides High minus low, people like to see moving averages:
close = df['Adj Close']
ma = pd.rolling_mean(close, 50)
print ma[-10:]

#so then let's bring back the high minus low, since ppl like to use that
# can also create new columns:
df['H-L'] = df.High - df.Low

HLMA = pd.rolling_mean(df['H-L'], 50)

print df.head()

ax1 = plt.subplot(2, 1, 1)
ax1.plot(close, label='sp500')
ax1.plot(ma, label='50MA')

ax2 = plt.subplot(2, 1, 2, sharex = ax1)
ax2.plot(df['H-L'], label='H-L')
ax2.plot(HLMA, label='HL 50MA')



The next tutorial:

  • Python and Pandas with Sentiment Analysis Database
  • Pandas Basics
  • Looking at our Data
  • Data Manipulation
  • Removing Outlier Plots
  • Basics for a Strategy
  • Dynamic Moving Averages
  • Strategy Function
  • Mapping function to dataframe
  • Beginning to back-test
  • More Analysis
  • Conclusion