danfo.getDummies
Convert categorical variable into dummy/indicator variables.
danfo.getDummies(kwargs)
Parameters | Type | Description | Default |
---|---|---|---|
data | Series or Dataframe | The data to dummify | â |
options | Object | These includes: columns: Array of column names to dummify. If not specified, all categorical columns are encoded. prefixSeparator: String separator for created columns e.g "_", prefix: Prefix for the new columns | { prefixSeparator: "-" } |
Node
Browser
const dfd = require("danfojs-node")
â
let datasf = ['pear', 'mango', "pawpaw", "mango", "bean"]
let sf1 = new dfd.Series(datasf)
â
let dum_df = dfd.getDummies(sf1, { prefix: "fruit" })
dum_df.print()
â
Output
ââââââââââââââ€ââââââââââââââââââââ€ââââââââââââââââââââ€ââââââââââââââââââââ€ââââââââââââââââââââ
â â fruit_pear â fruit_mango â fruit_pawpaw â fruit_bean â
ââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââ¢
â 0 â 1 â 0 â 0 â 0 â
ââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââ¢
â 1 â 0 â 1 â 0 â 0 â
ââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââ¢
â 2 â 0 â 0 â 1 â 0 â
ââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââ¢
â 3 â 0 â 1 â 0 â 0 â
ââââââââââââââŒâââââ âââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââ¢
â 4 â 0 â 0 â 0 â 1 â
ââââââââââââââ§ââââââââââââââââââââ§ââââââââââââââââââââ§ââââââââââââââââââââ§ââââââââââââââââââââ
Node
Browser
const dfd = require("danfojs-node")
â
let data = { fruits: ['pear', 'mango', "pawpaw", "mango", "bean"],
Count: [20, 30, 89, 12, 30],
Country: ["NG", "NG", "GH", "RU", "RU"]}
â
let df = new dfd.DataFrame(data)
df.print()
â
let dum_df = dfd.getDummies(df)
dum_df.print()
â
Output
âââââ€ââââââââââââââââââââ€ââââââââââââââââââââ€ââââââââââââââââââââ
â â fruits â Count â Country â
âââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââ¢
â 0 â pear â 20 â NG â
âââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââ¢
â 1 â mango â 30 â NG â
âââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââ¢
â 2 â pawpaw â 89 â GH â
âââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââ¢
â 3 â mango â 12 â RU â
âââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââ¢
â 4 â bean â 30 â RU â
âââââ§ââââââââââââââââââââ§ââââââââââââââââââââ§ââââââââââââââââââââ
â
â
//after dummification
â
âââââ€ââââââââââââââââââââ€ââââââââââââââââââââ€ââââââââââââââââââââ€ââââââââââââââââââââ€ââââââââââââââââââââ€ââââââââââââââââââââ€ââââââââââââââââââââ€ââââââââââââââââââââ
â â Count â fruits_pear â fruits_mango â ... â fruits_bean â Country_NG â Country_GH â Country_RU â
âââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââ¢
â 0 â 20 â 1 â 0 â ... â 0 â 1 â 0 â 0 â
âââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââ¢
â 1 â 30 â 0 â 1 â ... â 0 â 1 â 0 â 0 â
âââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââ¢
â 2 â 89 â 0 â 0 â ... â 0 â 0 â 1 â 0 â
âââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââ¢
â 3 â 12 â 0 â 1 â ... â 0 â 0 â 0 â 1 â
âââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââ¢
â 4 â 30 â 0 â 0 â ... â 1 â 0 â 0 â 1 â
âââââ§ââââââââââââââââââââ§ââââââââââââââââââââ§ââââââââââââââââââââ§ââââââââââââââââââââ§ââââââââââââââââââââ§ââââââââââââââââââââ§ââââââââââââââââââââ§ââââââââââââââââââââ
Node
Browser
const dfd = require("danfojs-node")
â
let data = { fruits: ['pear', 'mango', "pawpaw", "mango", "bean"],
Count: [20, 30, 89, 12, 30],
Country: ["NG", "NG", "GH", "RU", "RU"]}
â
let df = new dfd.DataFrame(data)
df.print()
â
let dum_df = dfd.getDummies(df, { columns: ['fruits']})
dum_df.print()
â
Output
âââââ€ââââââââââââââââââââ€ââââââââââââââââââââ€ââââââââââââââââââââ
â â fruits â Count â Country â
âââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââ¢
â 0 â pear â 20 â NG â
âââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââ¢
â 1 â mango â 30 â NG â
âââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââ¢
â 2 â pawpaw â 89 â GH â
âââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââ¢
â 3 â mango â 12 â RU â
âââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââ¢
â 4 â bean â 30 â RU â
âââââ§ââââââââââââââââââââ§ââââââââââââââââââââ§ââââââââââââââââââââ
â
â
//after dummification
â
âââââ€ââââââââââââââââââââ€ââââââââââââââââââââ€ââââââââââââââââââââ€ââââââââââââââââââââ€ââââââââââââââââââââ€ââââââââââââââââââââ
â â Count â Country â fruits_pear â fruits_mango â fruits_pawpaw â fruits_bean â
âââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââ¢
â 0 â 20 â NG â 1 â 0 â 0 â 0 â
âââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒâââââââââââââââââ âââŒââââââââââââââââââââŒââââââââââââââââââââ¢
â 1 â 30 â NG â 0 â 1 â 0 â 0 â
âââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââ¢
â 2 â 89 â GH â 0 â 0 â 1 â 0 â
âââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââ¢
â 3 â 12 â RU â 0 â 1 â 0 â 0 â
âââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââŒââââââââââââââââââââ¢
â 4 â 30 â RU â 0 â 0 â 0 â 1 â
âââââ§ââââââââââââââââââââ§ââââââââââââââââââââ§ââââââââââââââââââââ§ââââââââââââââââââââ§ââââââââââââââââââââ§ââââââââââââââââââââ
Last modified 1yr ago