Hướng dẫn return trong Python
Giả sử chúng ta có một hàm để hiển thị bình phương của một số được truyền vào lên màn hình:
def func(num):
print(num ** 2)
Trong trường hợp không cần hiển thị giá trị ngay lập tức lên màn hình, có thể ghi nó trước vào một biến nào đó:
res = func(3) # trong biến res bây giờ là 9
Để làm điều này, trong Python có một lệnh đặc biệt
return, cho phép
chỉ định giá trị mà hàm
trả về. Từ trả về
được hiểu là giá trị sẽ được ghi vào
biến, nếu gán hàm được gọi cho nó.
Hãy viết lại hàm của chúng ta để nó không xuất kết quả ra console, mà trả về kết quả đó vào một biến:
def func(num):
return num ** 2
Bây giờ hãy ghi kết quả thực thi hàm vào một biến:
res = func(3)
Sau khi dữ liệu được ghi vào biến, chúng có thể được, ví dụ, hiển thị lên màn hình:
res = func(3)
print(res) # sẽ hiển thị 9
Hoặc có thể sửa đổi dữ liệu này trước, rồi sau đó hiển thị chúng lên màn hình:
res = func(3)
res = res + 1
print(res) # sẽ hiển thị 10
Có thể ngay lập tức thực hiện một số hành động với kết quả của hàm trước khi ghi vào biến:
res = func(3) + 1
print(res) # sẽ hiển thị 10
Có thể gọi hàm của chúng ta nhiều lần cho các số khác nhau và cộng các kết quả của nó:
res = func(2) + func(3)
print(res) # sẽ hiển thị 13
Để rút gọn code đã viết, có thể ngay lập tức
gọi hàm, truyền nó vào tham số print:
print(func(3))
Tạo một hàm, nhận một số làm tham số,
và trả về lập phương của số đó. Sử dụng hàm này để tìm lập phương
của số 3 và ghi nó vào biến
res.
Sử dụng hàm bạn đã tạo để tìm
tổng lập phương của số 2 và số 3
và ghi nó vào biến res.