Python Data Science Toolbox Part 1 Learning 1 - User-defined functions

from:https://campus.datacamp.com/courses/python-data-science-toolbox-part-1/writing-your-own-functions?ex=1

• Strings in Python

To assign the string

company = 'DataCamp'

You've also learned to use the operations + and * with strings. Unlike with numeric types such as ints and floats, the + operator concatenates strings together, while the * concatenates multiple copies of a string together. In this exercise, you will use the + and * operations on strings to answer the question below. Execute the following code in the shell:

object1 = "data" + "analysis" + "visualization"
object2 = 1 * 3
object3 = "1" * 3

->object1 contains "dataanalysisvisualization", object2contains 3, object3 contains "111".

• Recapping built-in functions

• Assign str(x) to a variable y1: y1 = str(x)
• Assign print(x) to a variable y2: y2 = print(x)
• Check the types of the variables x, y1, and y2.

->x is a float, y1 is a str, and y2 is a NoneType.

It is important to remember that assigning a variable y2 to a function that prints a value but does not return a value will results in that variable y2 being oftypeNoneType.

• Write a simple function

You can use it as a pattern to define shout().

def square():
    new_value = 4 ** 2
    return new_value

# Define the function shout

def shout():

"""Print a string with three exclamation marks"""

# Concatenate the strings: shout_word

shout_word = 'congratulations'+ '!!!'

# Print shout_word

print(shout_word)

# Call shout

shout()

• Single-parameter functions

You will now update shout() by adding a parameter so that it can accept and process any string argument passed to it.

# Define shout with the parameter, word

def shout(word):

"""Print a string with three exclamation marks"""

# Concatenate the strings: shout_word

shout_word = str(word) + '!!!'

# Print shout_word

print(shout_word)

# Call shout with the string 'congratulations'

shout('congratulations')

• Functions that return single values

You're getting very good at this! Try your hand at another modification to the shout() function so that it now returns a single value instead of printing within the function. Recall that thereturn keyword lets you return values from functions.

# Define shout with the parameter, word

def shout(word):

"""Return a string with three exclamation marks"""

# Concatenate the strings: shout_word

shout_word = str(word) + '!!!'

# Replace print with return

return shout_word

# Pass 'congratulations' to shout: yell

yell = shout('congratulations')

# Print yell

print(yell)

Multiple parameters and return values

• Functions with multiple parameters

Here, you will modify shout() to accept two arguments.

# Define shout with parameters word1 and word2

def shout(word1,word2):

"""Concatenate strings with three exclamation marks"""

# Concatenate word1 with '!!!': shout1

shout1 = word1 + '!!!'

# Concatenate word2 with '!!!': shout2

shout2 = word2 + '!!!'

# Concatenate shout1 with shout2: new_shout

new_shout = shout1 + shout2

# Return new_shout

return new_shout

# Pass 'congratulations' and 'you' to shout(): yell

yell = shout('congratulations', 'you')

# Print yell

print(yell)

• A brief introduction to tuples

how to construct, unpack, and access tuple elements.

a, b, c = even_nums

# Unpack nums into num1, num2, and num3

print(nums)

num1, num2, num3 = (3, 4, 6)

# Construct even_nums

print(nums)

nums = (2, 4, 6)

print(nums)

even_nums = nums

print(even_nums)

• Function that return multiple values

Here you will return multiple values from a function using tuples.

Note that the return statement return x, y has the same result as return (x, y): the former actually packs x and yinto a tuple under the hood!

# Define shout_all with parameters word1 and word2

def shout_all(word1, word2):

# Concatenate word1 with '!!!': shout1

shout1 = word1 + '!!!'

# Concatenate word2 with '!!!': shout2

shout2 = word2 + '!!!'

# Construct a tuple with shout1 and shout2: shout_words

shout_words = (shout1, shout2)

# Return shout_words

return shout_words

# Pass 'congratulations' and 'you' to shout_all(): yell1, yell2

yell1, yell2 = shout_all('congratulations','you')

# Print yell1 and yell2

print(yell1)

print(yell2)

Bringing it all together

• Bringing it all together (1)

You've learned how to add parameters to your own function definitions, return a value or multiple values with tuples, and how to call the functions you've defined.

For this exercise, your goal is to recall how to load a dataset into a DataFrame. The dataset contains Twitter data and you will iterate over entries in a column to build a dictionary in which the keys are the names of languages and the values are the number of tweets in the given language. The file tweets.csv is available in your current directory.

# Import pandas

import pandas as pd

# Import Twitter data as DataFrame: df

df = pd.read_csv('tweets.csv')

# Initialize an empty dictionary: langs_count

langs_count = {}

# Extract column from DataFrame: col

col = df['lang']

# Iterate over lang column in DataFrame

for entry in col:

# If the language is in langs_count, add 1

if entry in langs_count.keys():

langs_count[entry] += 1

# Else add the language to langs_count, set the value to 1

else:

langs_count[entry] = 1

# Print the populated dictionary

print(langs_count)

Great job! You've now defined the functionality for iterating over entries in a column and building a dictionary with keys the names of languages and values the number of tweets in the given language.

• Bringing it all together (2)

In this exercise, you will define a function with the functionality you developed in the previous exercise, return the resulting dictionary from within the function, and call the function with the appropriate arguments

For your convenience, the pandas package has been imported aspd and the 'tweets.csv' file has been imported into thetweets_df variable.

# Define count_entries()

def count_entries(df, col_name):

"""Return a dictionary with counts of

occurrences as value for each key."""

# Initialize an empty dictionary: langs_count

langs_count = {}

# Extract column from DataFrame: col

col = df[col_name]

# Iterate over lang column in DataFrame

for entry in col:

# If the language is in langs_count, add 1

if entry in langs_count.keys():

langs_count[entry] += 1

# Else add the language to langs_count, set the value to 1

else:

langs_count[entry] = 1

# Return the langs_count dictionary

return langs_count

# Call count_entries(): result

result = count_entries(tweets_df,'lang')

# Print the result

print(result)