Skip to content

Instantly share code, notes, and snippets.

@dataman-git
dataman-git / fe_grunfeld
Last active March 16, 2020 23:19
fe_grunfeld
# Coding method 1
from linearmodels.panel import PanelOLS
import statsmodels.api as sm
exog = sm.add_constant(gf[['value','capital']])
grunfeld_fe = PanelOLS(gf['invest'], exog, entity_effects=True, time_effects=False)
grunfeld_fe = grunfeld_fe.fit()
print(grunfeld_fe)
# Coding method 2
grunfeld_fe = PanelOLS.from_formula("invest ~ value + capital + EntityEffects", data=gf)
# Coding method 1
from linearmodels.panel import PanelOLS
import statsmodels.api as sm
exog = sm.add_constant(gf[['value','capital']])
grunfeld_fet = PanelOLS(gf['invest'], exog, entity_effects=True, time_effects=True)
grunfeld_fet = grunfeld_fe.fit()
print(grunfeld_fet)
# Coding method 2
grunfeld_fet = PanelOLS.from_formula("invest ~ value + capital + EntityEffects + TimeEffects", data=gf)
@dataman-git
dataman-git / fatalities
Last active March 17, 2020 02:56
fatalities
import pandas as pd
import numpy as np
import seaborn as sns
fatalities = pd.read_csv('/fatality.csv')
fatalities.head()
avg = fatalities.groupby('year')['mrall','beertax'].mean()
avg
sns.set(style="darkgrid")
g = sns.jointplot("beertax", "mrall",
data=avg, kind="reg",
color="m", height=6)
@dataman-git
dataman-git / fatalities_fe
Last active March 17, 2020 02:14
fatalities_fe
# Coding method 1
from linearmodels.panel import PanelOLS
import statsmodels.api as sm
exog = sm.add_constant(fatalities[['beertax','mlda','jaild',
'comserd','vmiles','unrate','perinc']])
fe = PanelOLS(fatalities['mrall'], exog, entity_effects=True, time_effects=True)
fe = fe.fit()
print(fe)
# Coding method 2
@dataman-git
dataman-git / fe2
Created March 17, 2020 02:15
fe2
# Coding method 1
from linearmodels.panel import PanelOLS
import statsmodels.api as sm
exog = sm.add_constant(fatalities[['beertax','mlda','jaild','comserd',
'vmiles','unrate','perinc']])
fe2 = PanelOLS(fatalities['mrall'], exog, entity_effects=True, time_effects=False)
fe2 = fe2.fit()
print(fe2)
# Coding method 2
@dataman-git
dataman-git / fe2_0
Created March 17, 2020 02:17
fe2_0
# Coding method 1
from linearmodels.panel import PanelOLS
import statsmodels.api as sm
exog = sm.add_constant(fatalities[['beertax','mlda',
'jaild','comserd','vmiles','unrate','perinc']])
fe2 = PanelOLS(fatalities['mrall'], exog,
entity_effects=True, time_effects=False)
fe2 = fe2.fit()
print(fe2)
@dataman-git
dataman-git / poolOLS2
Created March 17, 2020 02:19
poolOLS2
# Coding method 1
from linearmodels.panel import PanelOLS
import statsmodels.api as sm
exog = sm.add_constant(fatalities[['beertax','mlda',
'jaild','comserd','vmiles','unrate','perinc']])
pooledOLS = PanelOLS(fatalities['mrall'], exog,
entity_effects=False, time_effects=False)
pooledOLS = pooledOLS.fit()
print(pooledOLS)